Linux下CUDA9.0和CUDA8.0切换记录及所遇问题

最新推荐文章于 2024-06-13 16:06:33 发布

进击的Kevin

最新推荐文章于 2024-06-13 16:06:33 发布

阅读量705

点赞数

分类专栏： CUDA

本文链接：https://blog.csdn.net/dongkaiwen48/article/details/89059341

版权

本文档记录了在Linux系统中如何在CUDA 8.0和CUDA 9.0之间进行切换，并解决了在切换过程中遇到的libcublas.so.9.0找不到的错误。通过修改环境变量、创建软链接以及运行ldconfig命令，最终成功解决问题。

摘要由CSDN通过智能技术生成

双CUDA版本切换及所遇问题## 标题

查看CUDA版本;nvcc -V
修改用户环境配置文件: vim ~/.bashrc 在文档的最后添加如图环境变量:

别忘记sudo source ~/.bashrc
cd /usr/local
ls 查看是否有cuda文件夹,如果没有执行sudo ln -s cuda-8.0 cuda(或者sudo ln -s cuda-9.0 cuda 取决于你想切换什么版本)
nvcc -V 查看版本,观察cuda版本是否变化
测试自己的程序,博主从8.0换到了9.0,在测试使用cuda9.0的项目时遇到了如下报错
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
解决方法:
检查 /usr/local/cuda-9.0/lib64 下是否有 libcublas.so.9.0
如果有的话则在终端输入sudo idconfig cuda-9.0/lib64
(关于ldconfig的相关知识参见:https://www.cnblogs.com/lyongde/p/4190588.html)
输入上述命令后又出现如下报错:
/sbin/ldconfig.real: /usr/local/cuda-9.0/lib64/libcudn

关注