nvidia-smi
右上角的cuda version:11.4,表示的是当前驱动支持的最高cuda版本是11.4,仍然需要自行手动安装cuda,所以我后续安装的版本是11.3
CUDA支持的显卡驱动版本
官网:1. CUDA 12.5 Release Notes — Release Notes 12.5 documentation (nvidia.com)
安装cuda
在官网安装时最好选择runfile,不建议选Network,因为官网默认给的代码里没有指定版本号,容易出现不兼容的情况
安装完成之后会显示
注意,以下的步骤是【必须的】,需要配置环境变量:
sudo vim ~/.bashrc
在内容的末尾添加内容,注意要将版本号替换成上面红色框中的版本号,不能直接复制我的内容
export PATH=/usr/local/cuda-11.3/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
最好更新配置文件
source ~/.bashrc
至此,才算完成了cuda toolkit 11.3的安装。
用nvcc --version 查看
参考( Ubuntu 20.04安装CUDA & CUDNN 手把手带你撸_ubuntu20.04安装cuda-CSDN博客)
CUDA Toolkit版本及可用PyTorch版本兼容关系(参考官网)