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