无法导入tensorflow-gpu

时间:2018-09-30 14:02:11

标签: python windows tensorflow

我尝试导入tensorflow-gpu,但使用不同版本的CUDA和cuDNN时遇到相同的错误。 我的GPU与CUDA兼容,我安装时没有问题,但是当我尝试导入tensorflow-gpu时,我得到了:

ImportError:DLL加载失败:没有任何附加说明。

无法加载本机TensorFlow运行时。

Tensorflow版本:1.11 CUDA版本:10 cudNN:7.3.1

我还尝试了cudaNN 5.0,因为我在一个对某些人有用的论坛上看到了。

2 个答案:

答案 0 :(得分:0)

假设您已经正确安装了nvidia驱动程序,请参考此表以查看应该使用哪个CUDA和cuDNN作为您喜欢的tensorflow版本。

tensorflow-cuda-cudnn-compatible

答案 1 :(得分:0)

尝试在cmd窗口中运行这些命令

SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin;%PATH%

SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\lib64;%PATH%

SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include;%PATH%

SET PATH=C:\tools\cuda\bin;%PATH%

还请参见Jialer Chew给出的图表,如果您现在正在使用tf版本2.2.0 cuda版本10.1和cudnn版本7.6.0。

相关问题