问题描述
pytorch 镜像位置：
https://hub.docker.com/r/pytorch/pytorch/tags
拉取镜像：
docker pull pytorch/pytorch:1.5-cuda10.1-cudnn7-runtime
1
查看本地现有镜像清单：
创建一个容器：
docker run --gpus all -td --name=liujiepeng_pytorch pytorch/pytorch bash
1
注意：其实这里有两个repository名字相同，tag名不同的版本。可以通过用repository:tag进行区分。
进入容器内部：
docker exec -it liujiepeng_pytorch bash
1
发现在内部找不到cuda和cudnn的位置？
可以看出，有CUDA的驱动(10.2)，但是找不到lib文件。也没有nvcc。但是在pytorch内部查看cuda版本：
可以看出cuda版本是10.1，这与拉取的镜像版本一致。
原因分析
究其原因是拉取的镜像是runtime版，所以只有runtime版本的cuda，并没有compiler版。这与安装带有CUDA libs的Pytorch二进制文件类似。Pytorch可以在GPU上运行，但是此时不需要安装完整的CUDA libs，所以也就不能在这个运行环境之外使用nvcc。
解决方案
如果拉取dev版：
docker pull pytorch/pytorch:1.5-cuda10.1-cudnn7-devel
1
则可以如下图看到lib文件的位置：
原文链接：https://blog.csdn.net/ljp1919/article/details/106209358
