查看镜像
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