1.首先安装Docker与Nvidia-docker，请参考链接：
Ubuntu18.04安装Nvidia-Docker2_思考实践的博客-CSDN博客
2.在docker中安装pytorch以及对应的cuda版本，登录dockerhub官网查看镜像，Docker Hub。如下图所示：
这里直接拉取最高的： 1.9.1-cuda11.1-cudnn8-devel镜像，选择devel版本的更方便。
使用docker pull命令下载镜像，会下载一段时间比较久。
3.镜像下载好之后，启动，使用如下命令：
nvidia-docker run -itd -p 5004:22 pytorch/pytorch:1.9.1-cuda11.1-cudnn8-devel
4.查询改容器的ID：docker ps
5.登录：
docker exec -it cb52a1489e3a /bin/bash
6.进入之后，使用命令conda init初始化conda
可以通过conda list 查看安装的环境，pip list显示不完全，我判断Docker官方给的镜像里面的环境大部分用Conda配置的
查看cuda版本信息以及pytorch使用显卡信息：
这不与你主机本地安装的Cuda版本冲突
7.后续还有在docker中安装jupyter notebook的操作可以参考下方链接最后一部分进行学习，目前没用上就不写了
参考资料
Ubuntu中，在docker中安装pytorch(gpu)+cuda-详细_管牛牛的博客-CSDN博客_docker中安装cuda
