近期在使用GPU时发现在命令行输入nvcc -V及nvidia -smi实时查看GPU时发现无法工作,但电脑已经装好CUDA。现给出解决办法。
1. nvcc -V
原因:CUDA已安装,但是没有添加到环境变量。
解决方法:
echo 'export PATH=/usr/local/cuda-10.1/bin/:$PATH'>>~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH'>>~/.bashrc
source ~/.bashrc
其中,/usr/local/cuda-10.1是本机CUDA的安装路径,大家需要根据自己的安装路径进行修改。添加完毕后打开bashrc可以看到添加了环境变量:
vim ~/.bashrc
看到文本最后出现以下内容:
#User specific aliases and functions
export PATH=/usr/local/cuda...
export LD_LIBRARY_PATH=/usr/local/cuda...
再输入nvcc -V即可看到nvidia cuda compiler的版本信息。
2. nvidia -smi
原因:多输入了个空格
解决办法:
直接输入 nvidia-smi,注意中间没有空格!!!