Ubuntu 20.04安装 CUDA 11.6

删除现有NVIDIA Driver

sudo apt-get --purge remove "*nvidia*"
sudo apt-get autoremove
sudo apt-get clean

禁用 Nouveau 驱动程序：

sudo bash -c "echo 'blacklist nouveau' > /etc/modprobe.d/blacklist-nouveau.conf"
sudo bash -c "echo 'options nouveau modeset=0' >> /etc/modprobe.d/blacklist-nouveau.conf"
sudo update-initramfs -u
sudo reboot

下载并安装cuda

下载cuda 11.6 链接

wget https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run
sudo sh cuda_11.6.2_510.47.03_linux.run

更新环境变量

echo 'export PATH=/usr/local/cuda-11.6/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc

echo 'export PATH=/usr/local/cuda-11.6/bin:$PATH' >> ~/.zshrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64:$LD_LIBRARY_PATH' >> ~/.zshrc

sudo sh -c 'echo "/usr/local/cuda-11.6/lib64" > /etc/ld.so.conf.d/cuda.conf'
sudo ldconfig

验证驱动程序安装

nvidia-smi
nvcc -V