本篇讲的重点是如何拉取带有cuda和cudnn的docker镜像,因此这些的镜像源的频道为NVIDIA:
官方地址:
https://hub.docker.com/r/nvidia/cuda
根据官方资料我们知道NVIDIA的docker的tag分为三类:
base版本、runtime版本、devel版本:
base版本只安装了cuda,runtime版本安装了cuda、cudnn、nccl,devel版本在runtime版本基础上又安装了头文件及开发工具。
个人建议还是安装devel版本,以免用到了没有出麻烦。
-------------------------------------------------------------------
指定操作为ubuntu22.04,指定cuda版本为12.2.0:
指定CPU架构为aarch64:
docker pull --platform=linux/aarch64 nvidia/cuda:12.2.0-devel-ubuntu22.04
指定CPU架构为x86_64:
docker pull --platform=linux/x86_64 nvidia/cuda:12.2.0-devel-ubuntu22.04
--------------------------------------------
==============================================
参考: