首页 > 解决方案 > 我可以在 docker 容器中使用降级的 cuda 版本吗?

问题描述

当我尝试在容器上运行降级的 cuda 版本时,会发生这种情况。在此处输入图像描述

我可以在 Container 上使用降级的 Cuda 版本和主机的另一个 cuda 版本吗?

标签: dockercudagpunvidianvidia-docker

解决方案


是的你可以。

当您指定 docker --gpus all(或该开关的某些变体)时,您正在使用支持 GPU 的 docker 版本(如果您使用的是 GPU,则建议这样做)。

正确设置后,该 docker GPU 启用应该允许您运行小于或等于nvidia-smi.

在您的情况下,nvidia-smi报告对 CUDA 版本 11.1 的支持,这意味着您可以拉/运行/使用基于 CUDA 10.0 的 docker 容器,例如(如您的示例中的情况。)

您显示的输出中似乎没有任何问题。

您在基础机器上安装的 CUDA 工具包版本与容器中使用的内容无关,但 CUDA 驱动程序版本(报告的nvidia-smi内容将决定您可以使用的最新 CUDA 工具包版本,无论是在容器中,或者在基础机器上。但是基础机器中的CUDA工具包和容器中的CUDA工具包不交互,并且彼此没有关联。

只有 GPU 驱动程序版本(即基础机器上安装的 GPU 驱动程序支持的 CUDA 版本)与兼容性有关。


推荐阅读