libcublas.so.10.0: cannot open shared object file: No such file or directory

最新推荐文章于 2024-07-10 17:01:26 发布

原创最新推荐文章于 2024-07-10 17:01:26 发布 · 2.7k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#cuda #cudnn #ubuntu

平台搭建专栏收录该内容

5 篇文章

订阅专栏

本文介绍了解决TensorFlow与CUDA版本不匹配的问题，包括如何正确设置环境变量以确保CUDA能够被TensorFlow识别，以及如何查找TensorFlow与CUDA版本之间的对应关系。

部署运行你感兴趣的模型镜像

报错的原因可能有几种，可逐一排除。

环境变量没有生效

安装cuda过程中需要配置环境变量：

#cuda路径更改成自己的实际cuda安装路径
export CUDA_HOME=/usr/local/cuda-10.0
export PATH=$PATH:$CUDA_HOME/bin 
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

如果只是在terminal执行上述代码，之后会失效，需要将上述代码写入~/.bashrc，才能保证一直生效。所以在根目录打开.bashrc：

vim ~/.bashrc

写入完成后保存更改：

source ~/.bashrc

tensorflow版本与cuda版本不对应

tensorflow版本与cuda版本是强相关的。
如果tensorflow版本需要的cuda版本是10.0，而系统内安装的是9.0，就会报错：找不到10.0相关的路径。
查看tensorflow版本与cuda版本对应关系：https://tensorflow.google.cn/install/source
截图示意：
在这里插入图片描述
如果是版本对应问题，更换版本吧~