docker pytorch-cuda 安装 devel 版本

查看镜像
docker image ls

docker 镜像文档查询链接

Docker Hubhttps://hub.docker.com/r/pytorch/pytorch/tags

选择自己合适的版本进行安装（直接复制即可获取安装代码）

拉镜像：

docker pull pytorch/pytorch:1.7.1-cuda11.0-cudnn8-devel

查看镜像
docker image ls

重新命名： 

docker tag 7554ac65eba5 jianhua/1.7.1-cuda11.0-cudnn8-devel

docker image ls

创建一个容器：

docker run --runtime=nvidia -it --ipc=host \
 -v /mnt/jianhua/:/home/ \
jianhua/1.7.1-cuda11.0-cudnn8-devel

 查看CUDA 路径:

echo $LD_LIBRARY_PATH 

 

CUDA 版本查看: nvcc -V