1.查看显卡驱动和支持的cuda版本
nvidia-smi
2.加入环境变量
gedit ~/.bashrc
复制下面这段
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.2/lib64
export PATH=$PATH:/usr/local/cuda-11.2/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.2
更新一下
source ~/.bashrc
检查版本,注意V大写
nvcc -V