搜索
您的当前位置:首页正文

安装Tensorflow提示Cuda版本不对

来源:二三娱乐

开启 GPU 支持的版本 (安装该版本的前提是已经安装了 CUDA sdk)$ pip install

安装完后在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不就行了嘛~

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.

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

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

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

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

Top