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 <<<
- .bashrc 环境变量配置文件在当前用户主目录下:
# 回到当前用户主目录
cd
vim .bashrc
把上面cuda的配置copy到最下面,指定你要使用的cuda的目录,即可
然后使用source命令使配置生效
source .bashrc
# source命令方式:
source ~/.bashrc
相关博文
Linux 可以安装多个版本的Cuda 吗 | 如何进行 Cuda 的版本切换
ubuntu18给当前用户安装cuda11.2 图文教程 | 配置cuDNN8.1