如何拉取指定CPU架构的并且指定ubuntu版本的并且指定cuda和cudnn版本的docker镜像

发布时间 2023-07-29 17:32:39作者: Death_Knight

本篇讲的重点是如何拉取带有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

 

--------------------------------------------

 

 

 

 

 

==============================================

 

 

参考: