以下说的内容为tensorflow与cuda和cudnn正确情况下,版本也对。下面为报错内容。
ImportError: libcublas.so.10.0: cannot open shared object file: No such file
查看环境变量
echo $PATH
发现有cuda的路径。
和在pycharm中查看环境变量
import os
os.system("echo $PATH")
发现这里没有cuda的路径。然后就有了两种解决方法
解决方法 1
sudo vi /etc/profile
里面添加相关的路径
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64
export PATH=$PATH:/usr/local/cuda-10.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.0
export LD_LIBRARY_PATH=/usr/local/cuda/include:$LD_LIBRARY_PATH
这种方法应该可以,但是我没成功。这样做确实在pycharm中看环境变量,有了cuda的。而libcublas.so.10.0文件在lib64中,也就是应该把lib64加入环境变量。可以试试。
解决方法 2
在file-》setting中。
添加
下边为添加内容
LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/include:$LD_LIBRARY_PATH
CUDA_HOME=/usr/local/cuda
参考
https://blog.csdn.net/tanhongguang1/article/details/81396724
https://blog.csdn.net/SJTUzhou/article/details/83931884
后记:
tensorrt报错
ImportError: libnvinfer.so.5: cannot open shared object file: No such file or directory
我在/usr/local/cuda/include下面找到了libnvinfer.so.5,但是添加了这个路径到环境变量,不灵。
ChengLaun 发布了118 篇原创文章 · 获赞 23 · 访问量 7万+ 私信 关注