可以看到安装过不同版本CUDA的路径。
接下来进行配置,编辑.bashrc文件:
vim ~/.bashrc
在最后加入两行,对应想使用的CUDA版本所在的路径,比如我想使用11.7版本:
export PATH=/usr/local/cuda-11.7/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64:$LD_LIBRARY_PATH
最后激活配置文件:
source ~/.bashrc
配置完成,可以用nvcc -V命令查看当前的CUDA(nvcc)版本。