安装显卡驱动
- 添加官方源并更新
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
- 查看推荐的驱动版本
ubuntu-drivers devices
- 安装推荐的版本
sudo apt instal nvidia-driver-510
- 使用nvidia-smi检查显卡信息
安装CUDA
-
查看nvidia驱动对应的CUDA版本:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/
-
选择对应版本的cuda:https://developer.nvidia.com/cuda-toolkit-archive
-
选择系统版本,根据提示进行下载
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda-repo-ubuntu2004-11-6-local_11.6.2-510.47.03-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-6-local_11.6.2-510.47.03-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-6-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
- 打开
~/.zshrc
,添加下列语句,保存并退出后运行source ~/.zshrc
export PATH=/usr/local/cuda/bin:${PATH}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH}
- 运行
nvcc --version
,判断是否安装成功