tensorflow的CUDA driver version is insufficient for CUDA runtime version 问题解决方案

喜欢ヅ旅行 2024-04-18 11:13 132阅读 0赞

CUDA的驱动程序版本跟CUDA的运行时版本不匹配!

1.CUDA driver version(驱动版本):就是NVIDIA GPU的驱动程序版本;

查看命令:nvidia-smi

70

我们看到我的GPU的驱动程序版本是:384.81

2.CUDA runtime version(运行时版本):是在python中安装的cudatoolkit和cudnn程序包的版本

查看命令:pip list 或者 conda list

70 1

python安装的cudatoolkit和cudnn程序包版本是:9.2

3.nvidia 驱动和cuda runtime 版本对应关系

运行时版本 驱动版本
CUDA 9.1 387.xx
CUDA 9.0 384.xx
CUDA 8.0 375.xx (GA2)
CUDA 8.0 367.4x
CUDA 7.5 352.xx
CUDA 7.0 346.xx
CUDA 6.5 340.xx
CUDA 6.0 331.xx
CUDA 5.5 319.xx
CUDA 5.0 304.xx
CUDA 4.2 295.41
CUDA 4.1 285.05.33
CUDA 4.0 270.41.19
CUDA 3.2 260.19.26
CUDA 3.1 256.40
CUDA 3.0 195.36.15

4.解决方案

从驱动和运行时的版本对应关系来看,版本为384.81的驱动程序对应的运行时版本是9.0,也就是说我们在python中安装cudatoolkit和cudnn程序包版本9.2是过高了。

因为系统中依赖GPU驱动的程序比较多,一般出现这种情况,我们都是更改cudatoolkit和cudnn程序包的版本

于是,先卸载python中安装cudatoolkit和cudnn程序包:pip uninstall cudnnpip uninstall cudatoolkit

然后安装对应版本的cudatoolkit和cudnn程序包:pip install cudatoolkit=9.0;pip install cudnn

之前用conda安装CUDA的,这里就要用:conda install cudatoolkit=9.0 #后面的数字是版本号,不唯一,按显卡对应的运行版本来。

5.为什么会出现这种情况呢:

  1. 一般出现这种情况是因为在python中安装tensorflowgpu版本时,pip会检查tensorflow依赖的其他的包,如果依赖的包没有安装,则会先安装最新版本的依赖包。这时候tensorflowgpu版本依赖cudatoolkitcudnn程序包,pip就会安装最新版本的cudatoolkitcudnn程序包,最终导致gpu驱动版本和cuda运行时版本不匹配。所以最好的办法是**先装相应版本的cudatoolkit再装tensorflow。**

————————————————
版权声明:本文为CSDN博主「li57681522」的原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/li57681522/article/details/82491617

发表评论

表情:
评论列表 (有 0 条评论,132人围观)

还没有评论,来说两句吧...

相关阅读