随笔-Ubuntu中多cuda版本环境配置

随笔-Ubuntu中多cuda版本环境配置

由于每个人做得任务不同，在实验室的服务器上，有多个用户和多个cuda版本，如何配置自己使用哪一个cuda呢？

看一下cuda的安装路径

• cuda一般安装在/usr/local/目录下
  

配置要用的版本的CUDA的路径

• 配置自己使用的cuda版本，编辑~/.bashrc

vi ~/.bashrc

• 插入以下配置。（想使用那个cuda版本就把配置中的cuda版本改成自己需要的那个）

# CUDA
export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

• 激活更改

source ~/.bashrc

补充：

• 查看cuda的版本

# cat /usr/local/cuda/version.txt
cat /usr/local/cuda-10.1/version.txt

• 查看cuDNN的版本。（这里查出来的是7.6.2）

# cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cat /usr/local/cuda-10.1/include/cudnn.h | grep CUDNN_MAJOR -A 2

• 查看gcc的版本(5.4.0)

gcc --version

补充

• 有多个cuda版本，配置一个默认的cuda。创建一个软连接：

ln -s cuda-10.1 cuda

• 再看cuda版本：

stat cuda

创作不易，喜欢的话加个关注点个赞，❤谢谢谢谢❤