ubuntu22 GTX1050TI / 1060安装驱动 cuda

Android_chunhui2024-05-19 20:59

安装驱动

bash 复制代码

# 卸载之前的驱动
sudo  apt-get remove --purge nvidia*
# 禁用系统默认驱动
sudo bash -c "echo blacklist nouveau > /etc/modprobe.d/blacklist-nvidia-nouveau.conf"
sudo bash -c "echo options nouveau modeset=0 >> /etc/modprobe.d/blacklist-nvidia-nouveau.conf"

结果如下

$ cat /etc/modprobe.d/blacklist-nvidia-nouveau.conf

blacklist nouveau

options nouveau modeset=0

查看可用版本并安装

bash 复制代码

sudo add-apt-repository ppa:graphics-drivers/ppa  
sudo apt-get update  
ubuntu-drivers devices
sudo apt-get install nvidia-driver-470
sudo reboot

使用nvidia-smi命令查看安装结果。

安装cuda

bash 复制代码

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.4.0/local_installers/cuda-repo-ubuntu2004-11-4-local_11.4.0-470.42.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.0-470.42.01-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-4-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
sudo reboot

配置环境变量

bash 复制代码

export CUDA_HOME=/usr/local/cuda-11
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}