解决安装tensorflow时,出现ImportError: libcublas.so.10.0,Failed to load the native TensorFlow runtime错误

最近在服务上安装tensorflow-gpu一直出现下面这个错误:
ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory
Failed to load the native TensorFlow runtime.
搞了半天,才发现是tensorflow-gpu和cuda版本不匹配。不同tensorflow-gpu版本对应着不同cuda和cuDNN版本如何查看cuda和cuDNN版本?
比如:
CUDA 8.0→cuDNN v5.1 / CUDA 8.0→cuDNN v6.0 / CUDA 9.0→cuDNN v7.0.5
tensorflow 1.6/1.5和CUDA 9.0对应,1.4/1.3和CUDA 8.0对应

于是乎,指定tensorflow-gpu版本重新安装(注意:不用先卸载之前安装tf,可直接pip,因为它在安装时会自动检测已经安装好的tf,并自动把它卸载掉)。

pip install  tensorflow-gpu==1.5

注:最好使用镜像比如清华镜像,速度差别不是一点点。

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

安装好之后可以测试一下

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

显示大段相关信息表示安装成功!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值