Linux 下 Cuda 环境变量配置——简记

Linux 下 Cuda 环境变量配置

一些代码运行的时候需要GPU来加速，cuda安装之后，或者使用root账户安装好的公用的cuda,当前用户需要在自己主目录下 .bashrc 正确配置要使用的cuda路径

• cuda配置如下：

#<<< config root's cuda  start <<<

export PATH=/usr/local/cuda-8.0/bin:$PATH
export CUDA_HOME=/usr/local/cuda-8.0/bin:$CUDA_HOME
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH

#<<< config root's cuda  end <<<

  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
 

• .bashrc 环境变量配置文件在当前用户主目录下：

# 回到当前用户主目录
cd 

vim .bashrc 
把上面cuda的配置copy到最下面，指定你要使用的cuda的目录，即可

  

 

  
1
  
2
  
3
  
4
  
5
 

然后使用source命令使配置生效

source  .bashrc
# source命令方式：
source ~/.bashrc

  

 

  
1
  
2
  
3
 

相关博文

Win10安装CUDA10和cuDNN

Linux 可以安装多个版本的Cuda 吗 | 如何进行 Cuda 的版本切换

ubuntu18给当前用户安装cuda11.2 图文教程 | 配置cuDNN8.1

文章来源: positive.blog.csdn.net，作者：墨理学AI，版权归原作者所有，如需转载，请联系作者。

原文链接：positive.blog.csdn.net/article/details/87894839