Linux下切换cuda版本 |
您所在的位置：网站首页 › linux切换系统 › Linux下切换cuda版本 |
Linux下切换cuda版本
|
背景：
在使用服务器跑深度学习代码时往往需要依据不同项目要求配置环境，这就需要不同的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版本
|
【本文地址】
今日新闻 |
推荐新闻 |