安装了NVIDIA驱动程序,同时也安装了CUDA,但使用nvcc -V
命令提示不存在,如下所示:
1 Command 'nvcc' not found, but can be installed with:
2 sudo apt install nvidia-cuda-toolkit
知识补充:为什么要使用nvcc命令?
使用nvcc -V
命令可以查看CUDA的版本,如下所示为正常的输入、输出内容,可以看出通过nvcc -V
命令,可以看到目前所使用的CUDA版本。
解决方案
根据提示内容,说nvcc
没有找到,但可以通过sudo apt install nvidia-cuda-toolkit
命令安装,注意:千万不能使用sudo apt install nvidia-cuda-toolkit
命令,否则会重新安装CUDA(而且会安装NVIDIA驱动可以支持的最高版本的CUDA)!!!
其实CUDA已经安装好了,只是没有在环境变量中配置,所以找不到该命令,需要做如下操作:
1)查看CUDA的位置:
1 cd /usr/local
一般在/usr/local目录下会有CUDA文件夹或软链
2)打开、编辑环境变量的配置文件:
1 vim ~/.bashrc
在文件末尾添加:
# cuda
export LD_LIBRARY_PATH=/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
然后按:wq保存退出编辑模式。
3)刷新~/.bashrc文件使其生效:
1 source ~/.bashrc
4)重新使用nvcc命令查看CUDA版本:
输入:
1 nvcc -V
输出:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Tue_Jul_11_02:20:44_PDT_2023
Cuda compilation tools, release 12.2, V12.2.128
Build cuda_12.2.r12.2/compiler.33053471_0
成功!