一、启动gpu-docker容器
docker run --name szhq-ssh -i -t -d --runtime nvidia -e NVIDIA_VISIBLE_DEVICES=1 tensorflow/tensorflow:test-gpu /bin/bash
## NVIDIA_VISIBLE_DEVICES=1 表示将GPU第1号核心分配给test-gpu容器使用。
二、安装openssh
apt-get update -y
apt-get install -y openssh-server openssh-client vim
三、配置SSHD
# passwd //记录密码，提供给使用者。
# vim ~/.bashrc
添加/usr/sbin/sshd
# mkdir -p /var/run/sshd
# vim /etc/ssh/sshd_config
PermitRootLogin yes
四、提交容器成为新的镜像
# docker commit 3ea7a99a0025 sshd_gpu_01
五、停止并删除容器
# docker stop [container-id]
# docker rm [container-id]
六、后台启动一个新的容器
# docker run --name szhq-gpu -d --runtime nvidia -e NVIDIA_VISIBLE_DEVICES=1 -p 8001:22 sshd_gpu_01 /usr/sbin/sshd -D
七、ssh远程登录该容器
# ssh root@localhost -p 8001
