随笔-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

在这里插入图片描述

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

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐