我已经在CentOS
上安装了cuda-8.0
和cudnn5.1
。然后,当导入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
但是,当我在文件夹 /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
中运行./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
使用 "pip3 list |grep tensorflow" 如果是 tensorflow-gpu (1.5.0( 版本,则所需的 cuda 版本是 9.0 和 cuDNN v7。
有关更多详细信息,请查看以下链接:https://github.com/tensorflow/tensorflow/releases
Tensorflow 安装指南需要更新。
我遇到了同样的问题。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.0 旁边安装 CUDA 9.1(这对我有用(
2.( 从 git 源代码自己构建 Tensorflow
无论哪种方式,都不要忘记将 PATH 变量添加到您的操作系统中,否则您会收到来自 python 解释器的问题中所述的错误消息。