我在cuda-8.0
上安装了cudnn5.1
和CentOS
。然后,在导入tensorflow
(python 3.6
)时,会出现上述错误。
我已在/ etc / profile中设置符号链接,如下所示。有没有人发生过这种问题?
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH
另外,令我困惑的是,我运行nvcc -V,它显示
Cuda compilation tools, release 8.0, V8.0.61
但是,当我在./deviceQuery
文件夹/usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
上运行device 0: "Tesla M40"
时,会显示
CUDA Driver Version / Runtime Version 9.1 / 8.0
deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 9.1, CUDA Runtime Version = 8.0, NumDevs = 1, Device0 = Tesla M40
答案 0 :(得分:0)
使用“pip3 list | grep tensorflow”检查你的tensorflow版本如果是版本tensorflow-gpu(1.5.0)那么所需的cuda版本是9.0和cuDNN v7。
请查看以下链接以获取更多详细信息: https://github.com/tensorflow/tensorflow/releases
需要更新Tensorflow安装指南。
答案 1 :(得分:0)
我遇到了同样的问题。 Tensorflow 1.5.0预编译为CUDA 9.0(已过时; 2017年9月)。
最新的CUDA版本是CUDA 9.1(2017年12月),sudo pip install tensorflow-gpu不适用于最新的CUDA 9.1。该问题有两种解决方案:
1。)在CUDA 9.1旁边安装CUDA 9.0(这对我有用)
2.。)自己从git源代码构建Tensorflow
无论哪种方式都不要忘记将PATH变量添加到您的操作系统,否则您将收到python解释器中问题中声明的错误消息。