ubuntu16.4+pycharm使用tensorflow-gpu版本出现的问题

错误

运行项目时,报错

ImportError:libcublas.so.9.0:cannot open shared object file:No such file or dircetory

设置环境变量和动态链接库,在命令行输入: 

sudo gedit /etc/profile 

在打开的文件末尾加入:

export PATH=/usr/local/cuda-9.0/bin:$PATH

创建链接文件,在命令行输入: 

sudo gedit /etc/ld.so.conf.d/cuda.conf 

在打开的文件中添加如下语句:

/usr/local/cuda/lib64

最后使链接立即生效: 

sudo ldconfig

报错

/sbin/ldconfig.real: /usr/local/cuda-9.0/lib64/libcudnn.so.7 不是符号链接

解决方法:

/usr/local/cuda/lib64$ sudo rm libcudnn.so.7
/usr/local/cuda/lib64$ sudo ln libcudnn.so.7.0.5 libcudnn.so.

猜你喜欢

转载自blog.csdn.net/zhngyue123/article/details/105367126