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

2022-09-02 20:06:14 浏览数 (1)

可能原因,CUDA版本不对

1、查看下版本信息

代码语言:javascript复制
cat /usr/local/cuda/version.txt  

2、根据CUDA和tensorflow版本对应的关系表,检查自己的版本是否匹配

3、如果不匹配

请安装相应的TensorFlow或者CUDA版本

安装对应版本TensorFlow的GPU版本

代码语言:javascript复制
pip install tensorflow-gpu==版本号

安装对应版本的CUDA

参考:https://blog.csdn.net/wf19930209/article/details/81879514

4、如果匹配,仍然报错

那是因为你的cuda环境变量配置有误,请执行:

代码语言:javascript复制
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/【CUDA版本】/lib64

例如

代码语言:javascript复制
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64

0 人点赞