ubuntu20.04安装nvidia驱动和cuda以及pytorch安装

nvidia驱动安装

下载驱动

根据不同的显卡型号选择下列命令，为了减少冲突，我选择了535版本较低的驱动。
执行sudo apt install -y nvidia-driver-535

控制nouveau驱动

由于nouveau驱动的原因，会和nvidia产生冲突，
故创建黑名单封锁nouveau：

sudo tee /etc/modprobe.d/blacklist-nouveau.conf <<EOF
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
EOF

然后：sudo update-initramfs -u
更新内核模块配置
最后sudo reboot

cuda安装

由于我是为了d2l的学习，所需要的python版本是3.9，且上面的nvidia驱动安装的是535，所以我选择了11.7版本的cuda

首先安装依赖

sudo apt update
sudo apt upgrade -y
sudo apt install -y build-essential gcc g++ make

然后通过wget下载：wget -c https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda-repo-ubuntu2004-11-7-local_11.7.0-515.43.04-1_amd64.deb
这里的-c参数是因为cuda较大，可供暂停

安装 CUDA 11.7

bash

安装下载的 deb 包

sudo dpkg -i cuda-repo-ubuntu2004-11-7-local_11.7.0-515.43.04-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-11-7-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt update
sudo apt install -y cuda-11-7

配置环境变量
打开 .bashrc 文件：
nano ~/.bashrc
在文件末尾添加以下内容：

export PATH=/usr/local/cuda-11.7/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

使配置生效：
source ~/.bashrc

验证安装

运行以下命令验证：
nvcc --version
若安装成功，会显示类似以下信息：

plaintext
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Wed_Jun__8_16:49:14_PDT_2022
Cuda compilation tools, release 11.7, V11.7.99
Build cuda_11.7.r11.7/compiler.31442593_0