docker容器中的cuda版本可以与主机不同吗?

时间:2018-05-18 09:22:15

标签: cuda

我想运行用cuda8.0和tensorflow-1.2编写的代码。我想在另一台机器上运行代码,其中cuda是9.0,所以我使用docker生成一个容器,使用这个命令:

FROM nvidia/cuda:8.0-cudnn5-devel

当我在此容器中运行代码时,会出现以下错误:

CUDA driver version is insufficient for CUDA runtime version

所以,这是错的吗?怎么解决?

1 个答案:

答案 0 :(得分:1)

对于任何偶然发现此问题的人,这里的问题是主机上的CUDA驱动程序库的版本对于容器内的CUDA运行时库而言太旧了。 CUDA驱动程序构成主机上NVIDIA驱动程序安装的一部分。

CUDA驱动程序不向前兼容,因此主机至少必须与您要在容器中使用的CUDA运行时版本相同。对于最广泛的选项,您应该将主机上的NVIDIA驱动程序和CUDA驱动程序更新为最新的稳定版本。