本文系转载,出处:https://blog.csdn.net/u011622208/article/details/113650011

Description
CUDA图像有三种风格,可以通过NVIDIA公共集线器存储库获得。

基础(base):从CUDA 9.0开始,包含了部署预构建CUDA应用程序的最低限度(libcudart)。
如果你想手动选择你想要安装的CUDA包,请使用这个映像。

运行时(runtime):通过添加CUDA工具包中的所有共享库扩展基本映像。
如果您有一个使用多个CUDA库的预构建应用程序,请使用此图像。

devel:通过添加编译器工具链、调试工具、头文件和静态库来扩展运行时映像。

使用这个图像从源代码编译一个CUDA应用程序。

例子

Running an interactive CUDA session isolating the first GPU

docker run -ti --rm --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=0 nvidia/cuda

Querying the CUDA 7.5 compiler version

docker run --rm --runtime=nvidia nvidia/cuda:7.5-devel nvcc --version

总结

runtime的包,没有cuda的编译工具nvcc

docker run --rm nvidia/cuda:11.1-runtime-ubuntu16.04 nvcc --version

docker: Error response from daemon: OCI runtime create failed: container_linux.go:370: starting container process caused: exec: "nvcc": executable file not found in $PATH: unknown.

devel的,是有cuda的nvcc包的

docker run --rm nvidia/cuda:11.1-cudnn8-devel-ubuntu16.04 nvcc --version

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Oct_12_20:09:46_PDT_2020
Cuda compilation tools, release 11.1, V11.1.105
Build cuda_11.1.TC455_06.29190527_0
Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐