Linux下切换cuda版本

最新推荐文章于 2024-08-16 20:07:11 发布

宿醉之星

最新推荐文章于 2024-08-16 20:07:11 发布

阅读量8.4k

点赞数 20

文章标签： linux

本文链接：https://blog.csdn.net/Nirvana15/article/details/119062355

版权

该博客介绍了如何在Linux环境中进行CUDA版本的切换，主要步骤包括编辑.bashrc文件，通过创建软链接或定义切换函数来实现不同CUDA版本的快速选择。此外，还强调了更新bashrc配置并验证CUDA版本的重要性。

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

背景：

在使用服务器跑深度学习代码时往往需要依据不同项目要求配置环境，这就需要不同的cuda版本

问题描述：

一般来说CUDA安装在 /usr/local 目录下。这里默认已经安装完毕。

解决方案：

1.在自己的环境下打开.bashrc文件，可使用vim ~/.bashrc命令；

2.1将下面几段代码粘贴进去：

这里可建立软连接，可以通过修改文件中的路径切换cuda版本。

# 再将下面三句话粘贴进去
export PATH="$PATH:/usr/local/cuda-8.0/bin"
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64/"
export LIBRARY_PATH="$LIBRARY_PATH:/usr/local/cuda-8.0/lib64"

2.2如果上一种方法失败，还有一种是在最后加入如下的切换函数：


#写入切换函数
#*******   switch cuda
function _switch_cuda {
   v=$1
   export PATH=/usr/local/cuda-$v/bin:$PATH
   export CUDADIR=/usr/local/cuda-$v
   export CUDA_HOME=/usr/local/cuda-$v
   export LD_LIBRARY_PATH=/usr/local/cuda-$v/lib64:$LD_LIBRARY_PATH
   nvcc --version
}
_switch_cuda 11.0  #在此更改版本
#*******

3.最后注意，一定要使用更新bashrc命令 source ~/.bashrc

4.使用nvcc --version命令查看cuda版本