my dockerfile: nvidia/cuda:11.3.0-cudnn8-devel-ubuntu18.04

Dockerfile

FROM nvidia/cuda:11.3.0-cudnn8-devel-ubuntu18.04
RUN mkdir -p /home/root
RUN apt update

RUN apt upgrade -y
RUN apt install -y apt-utils dialog
RUN apt install -y bash-completion locate vim tmux unzip htop
RUN apt install -y openssh-server # net-tools iputils-ping curl
RUN apt install -y build-essential pkg-config gdb git
RUN apt install -y libssl-dev libgoogle-glog-dev libgflags-dev

RUN touch /bin/init
RUN echo "#!/bin/bash" >> /bin/init
RUN echo "service ssh start" >> /bin/init
RUN echo "while true; do sleep 1; done;" >> /bin/init
RUN chmod +x /bin/init

RUN echo "PermitRootLogin yes" >> /etc/ssh/sshd_config
RUN echo "root:root" | chpasswd

RUN updatedb