docker配置pytorch开发环境

1、ubuntu安装docker并pull pytorch的镜像。

2、启动镜像时注意端口和文件夹的映射

docker run -it -p 10022:22 -p 16006:6006 -p 18888:8888 -v /user/project:/home/user/disk2/menfer/project --name menfer pytorch/pytorch:1.2-cuda10.0-cudnn7-devel

重点是docker开启新镜像后的操作，一般需要配置jupyter，ssh等。

3、更新apt-get并安装vim

apt-get update

apt-get install vim

4、更改apt源为国内源（参考https://blog.csdn.net/zgljl2012/article/details/79065174）

vim /etc/apt/sources.list

deb http://mirrors.ustc.edu.cn/ubuntu/ xenial main restricted universe multiverse

deb http://mirrors.ustc.edu.cn/ubuntu/ xenial-security main restricted universe multiverse

deb http://mirrors.ustc.edu.cn/ubuntu/ xenial-updates main restricted universe multiverse

deb http://mirrors.ustc.edu.cn/ubuntu/ xenial-proposed main restricted universe multiverse

deb http://mirrors.ustc.edu.cn/ubuntu/ xenial-backports main restricted universe multiverse

deb-src http://mirrors.ustc.edu.cn/ubuntu/ xenial main restricted universe multiverse

deb-src http://mirrors.ustc.edu.cn/ubuntu/ xenial-security main restricted universe multiverse

deb-src http://mirrors.ustc.edu.cn/ubuntu/ xenial-updates main restricted universe multiverse

deb-src http://mirrors.ustc.edu.cn/ubuntu/ xenial-proposed main restricted universe multiverse

deb-src http://mirrors.ustc.edu.cn/ubuntu/ xenial-backports main restricted universe multiverse

apt-get update

5、安装ssh

apt-get install openssh-server

6、开启ssh并设置允许root登录（参考https://blog.csdn.net/wy_97/article/details/78294562）

service ssh start

测试ssh可以正常启动

然后更改允许root登录：

vim /etc/ssh/sshd_config

更改PermitRootLogin为yes

重启ssh：service ssh restart

7、更改root密码并使用ssh软件登录

passwd

8、在使用ssh登录docker的时候，可能会出现使用命令的时候报command not found的错误，而不使用ssh直接用docker则可以用，这是因为ssh和docker本身的PATH不同步，可以使用export命令在docker中查看PATH，然后复制PATH后的内容xxx，在ssh中使用export PATH=xxx来解决，之后就可以在ssh使用python等命令了。