Docker 安装PyTorch + jupyter

一、安装容器

1.方便起见，直接pull一个pytorch镜像，anibali/pytorch，选择合适的cuda版本。

$ docker pull anibali/pytorch:1.5.0-cuda10.2

参考https://github.com/anibali/docker-pytorch（注意看文档，他只在Ubuntu上测试过！）

2.上述文档中给出的usage中是这样的。

$ docker run --rm -it --init \
  --gpus=all \
  --ipc=host \
  --user="$(id -u):$(id -g)" \
  --volume="$PWD:/app" \
  anibali/pytorch python3 main.py

这段命令中镜像并没有指定版本，但最好指定，因为pull 镜像时加上版本了，如果这个地方不加版本它会创建一个新的镜像，跟原来的tag不同而已，像下面这样

如果你本身就有写好的python工程文件，可以修改--volume参数将文件夹挂载到容器中，然后直接使用这段命令执行。退出容器后容器会自行删除。

但是我需要在这个容器中安装jupyter lab，方便学习使用。故我创建container的命令是

$ docker run -it --init   --gpus=all   --ipc=host  --name pytorch -p 1778:8888  --volume="$PWD:/app"   anibali/pytorch python3

直接进入python 3环境中进行测试。可参考

https://blog.csdn.net/weixin_35576881/article/details/89709116

这里的-p 端口映射是因为服务器上的8888端口已经被其他小伙伴使用了，所以随便选择了一个端口。如果你的8888端口没有被占用，也可以使用8888，比较好记。

二、安装JupyterLab

接下来在容器中安装JupyterLab。

# 开启容器
docder start  containername

# 以交互式进入容器
docker exec -it containername /bin/bash

以下在容器内部进行

# 安装jupyterlab
conda install -c conda-forge jupyterlab

# 启动jupyterlab后远程浏览器登陆
jupyter lab --ip=0.0.0.0 --no-browser --allow-root

然后你会发现可以使用python 3内核运行程序，但你没办法访问文件，创建新的文件，提示permission denied

这是因为我们之前选择了--volume参数，容器挂载了一个外部的目录存放我们的文件，方便数据在容器内外和不同容器之间共享，这个文件夹在容器内部是没有权限去访问的。所以可以

1.暴力的chmod 777  修改权限（最好别这样，尤其是一台服务器多人使用时）；

2.或参考这篇文章修改UIDhttps://blog.csdn.net/queena_qing/article/details/86072084，如果你要在其他container中使用这个文件夹，最好修改容器中用户的UID而不是像这个博主一样修改实际的文件属主UID