Dockerfile基础镜像选择
Dockerfile的最开始都是From xxx,表示选择的基础镜像是什么版本,可以在nvidia的gitlab上选择版本。需要注意nvidia/cuda:latest的表示方式已经被弃用,可以看前面写的文章
例如
1 | |
其中10.0是指cuda的版本
cudnn7指与cuda10.0对应的cudnn版本
devel指镜像中的nvcc包包含cuda
ubuntu16.04指镜像中ubuntu的版本
其中,除了ubuntu版本可以与本地不同之外,其余的都要和主机上的版本对应,否则运行docker将无法使用本地的GPU硬件。
ubuntu版本能选最高就选最高,因为低版本能安装的库版本也很低,无法使用一些新功能
参考
https://gitlab.com/nvidia/container-images/cuda/blob/master/doc/supported-tags.md
https://blog.csdn.net/u011622208/article/details/113650011
https://blog.csdn.net/YoJayC/article/details/120209482?spm=1001.2014.3001.5501
Dockerfile基础镜像选择
http://yojayc.github.io/2021/09/11/Dockerfile基础镜像选择/