cuda8.0如何安装TensorFlow-gpu?
How to install TensorFlow-gpu with cuda8.0?
我尝试按照官网的说明安装,导致导入tensorflow时出现ImportError:
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
我运行代码cat /usr/local/cuda/version.txt
,显示我的cuda版本是8.0.61.
好像是tensorflow在找cuda 9.0。我无法升级 cuda,因为我在共享 gpu-server 上工作,而且我没有 root 权限。
有什么办法可以让tensorflow在cuda 8.0上运行吗?或者还有其他可用的方式吗?
谢谢!!
您需要为 CUDA-8 安装 1.4.1
版本
pip install tensorflow-gpu==1.4.1
最新的(版本 1.5
)适用于 CUDA-9
我遇到了类似的问题,直到我找到
https://www.tensorflow.org/install/install_sources#tested_source_configurations
检查你安装的cuda版本和cudnn版本,然后找出哪个版本的tensorflow-gpu与上面提到的link兼容。
我已经安装了 cuda 8 和 cudnn v5.1,因此通过上面的检查 link tensorflow-gpu 1.2.0 是兼容的,安装后使用
pip install tensorflow-gpu==1.2.0
对我有用。
我尝试按照官网的说明安装,导致导入tensorflow时出现ImportError:
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
我运行代码cat /usr/local/cuda/version.txt
,显示我的cuda版本是8.0.61.
好像是tensorflow在找cuda 9.0。我无法升级 cuda,因为我在共享 gpu-server 上工作,而且我没有 root 权限。
有什么办法可以让tensorflow在cuda 8.0上运行吗?或者还有其他可用的方式吗?
谢谢!!
您需要为 CUDA-8 安装 1.4.1
版本
pip install tensorflow-gpu==1.4.1
最新的(版本 1.5
)适用于 CUDA-9
我遇到了类似的问题,直到我找到
https://www.tensorflow.org/install/install_sources#tested_source_configurations
检查你安装的cuda版本和cudnn版本,然后找出哪个版本的tensorflow-gpu与上面提到的link兼容。
我已经安装了 cuda 8 和 cudnn v5.1,因此通过上面的检查 link tensorflow-gpu 1.2.0 是兼容的,安装后使用
pip install tensorflow-gpu==1.2.0
对我有用。