CUDA我爱编程

安装Tensorflow提示Cuda版本不对

2016-12-27  本文已影响676人  Dorts

今晚给实验室机器安装tensorflow,根据 TF中文社区 http://www.tensorfly.cn/tfdoc/get_started/os_setup.html 相关教程,用pip如下命令安装:

开启 GPU 支持的版本 (安装该版本的前提是已经安装了 CUDA sdk)$ pip install https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.5.0-cp27-none-linux_x86_64.whl

安装完后在python中导入tensorflow竟然提示找不到Cuda7.0,具体如下:
[TensorFlow: libcudart.so.7.0: cannot open shared object file: No such file or directory

马丹,老子机器上明明是cuda7.5的版本,torch都能跑起来。

难道必须下载cuda7.0嘛?
经过仔细分析,才发现原来 TF 中文社区的命令下载的tensorflow版本太低了!tensorflow-0.5.0,竟然是5.0的版本。
我立马去TF官网查看安装手册,发现英文社区已经到了tensorflow-0.12.0了!果然尼玛中文社区太坑了。不过12.0版本的TF要求的CUDA至少是8.0.
分析到这我就有谱了,我直接下载一个满足cuda7.5版本的Tensorflow不就行了嘛~

于是在连接http://wiki.jikexueyuan.com/project/tensorflow-zh/get_started/os_setup.html中找到了相应的下载命令:

Ubuntu/Linux 64-bit, GPU enabled, Python 2.7. Requires CUDA toolkit 7.5 and CuDNN v4.

For other versions, see "Install from sources" below.

$ sudo pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.8.0-cp27-none-linux_x86_64.whl

运气真是好呀~原来8.0的版本可以用CUDA7.5支持。

果然先卸载低版本再安装这个版本,然后再Ipython中就能跑起来了!

不需要重新安装很麻烦的7.0的cuda了!

总结:中文资料就是个坑,尼玛!

上一篇下一篇

猜你喜欢

热点阅读