ImportError: libcudnn.so.6: cannot open shared object file: No such file or directory

今天在安装TensorFlow1.4.1,安装完成后运行爆出如上错误,这是因为1.3以后的版本需要cuDNN v6.0做支持了,本机上安装的是cuDNN v8.0。看网上教程说要安装低版本的TensorFlow,作者感觉不用,就做了如下尝试。

解决方法如下:

1.先到英伟达的官网下载cudnn-8.0-linux-x64-v6.0的lib包。下载完后是cudnn-8.0-linux-x64-v6.0.solitairetheme8这样的文件。

2.解压该文件:

	 cp  cudnn-8.0-linux-x64-v5.1.solitairetheme8 cudnn-8.0-linux-x64-v5.1.tgz
	tar -xvf cudnn-8.0-linux-x64-v5.1.tgz

3.复制文件到/usr/local/cuda/lib64文件夹下,并给相应的权限

	sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ 
	sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

再运行 import tensorflow as tf 完美解决。


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值