背景:
在使用服务器跑深度学习代码时往往需要依据不同项目要求配置环境,这就需要不同的conda版本
问题描述:
一般来说CUDA安装在 /usr/local 目录下。这里默认已经安装完毕。
?
解决方案:
1.在自己的环境下打开.bashrc文件,可使用vim ~/.bashrc命令;
2.1将下面几段代码粘贴进去:
这里可建立软连接,可以通过修改文件中的路径切换conda版本。
# 再将下面三句话粘贴进去
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命令查看conda版本
|