使用root用户安装NVIDIA显卡驱动+cuda+cudnn后,将环境变量加入到/etc/profile中,然后source /etc/profile使其生效,此时,root用户执行
nvcc -V
看到cuda已经可以生效
如果其他用户想要使用该cuda,只需要在普通用户的~/.bashrc文件中加入
cuda的环境变量
export PATH=/usr/local/cuda-11.6/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
或者(推荐下者)
export PATH=${PATH}:/usr/local/cuda-11.6/bin
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda-11.6/lib64
再执行source ~/.bashrc使其生效即可