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

Lil*_*hen 9 cuda

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

FROM nvidia/cuda:8.0-cudnn5-devel
Run Code Online (Sandbox Code Playgroud)

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

CUDA driver version is insufficient for CUDA runtime version
Run Code Online (Sandbox Code Playgroud)

那么,有错吗?如何解决?

Ed *_*din 9

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

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