背景:

在使用服务器跑深度学习代码时往往需要依据不同项目要求配置环境,这就需要不同的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版本

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐