Ubuntu/Linux 安装CUDA、检测（PyTorch准备）

先安装驱动

Ubuntu/Linux 安装GPU 驱动&检测（PyTorch准备）

更新Ubuntu

sudo apt update
sudo apt upgrade 

检查PyTorch官网支持CUDA版本

https://pytorch.org/get-started/locally/#windows-package-manager

根据PyTorch选择CUDA版本11.8 以下是Ubuntu22.04 + CUDA11.8

https://developer.nvidia.com/cuda-11-8-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=deb_local

安装CUDA11.08

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda-repo-ubuntu2204-11-8-local_11.8.0-520.61.05-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2204-11-8-local_11.8.0-520.61.05-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2204-11-8-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda

重启

sudo reboot now # 重启

设置路径

https://zhuanlan.zhihu.com/p/630067986

echo 'export PATH=/usr/local/cuda-11.8/bin:$PATH' >> ~/.bashrc
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}} # 或者是这种路径设置
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
sudo ldconfig

检查

nvcc -V

遇到问题

重启修复

(Ubuntu 开机后显示失败)[https://blog.csdn.net/yingmohuanzhou/article/details/138168254]

卸载

sudo apt purge nvidia* -y
sudo apt remove nvidia-* -y
sudo rm /etc/apt/sources.list.d/cuda*
sudo apt autoremove -y && sudo apt autoclean -y
sudo rm -rf /usr/local/cuda*

重启

reboot now

重新安装驱动

Ubuntu/Linux 安装GPU 驱动&检测（PyTorch准备）