Ubuntu18.04 下 CUDA 11.4和 Pytorch 11.4安装

参考 最全：在Ubuntu 18安装Nvidia Driver, Cuda 10.2, Pytorch及一些疑问的解答和讨论 - 知乎  和 Ubuntu搭建Pytorch，就这一篇就够了_Chengyunlai的博客-CSDN博客_ubuntu安装pytorch

CUDA 11.4

为了后面顺利进行，确保系统不存在Nvidia Driver，Cuda相关文件。
 

sudo apt-get purge nvidia*
sudo apt --purge remove "cublas*" "cuda*"
sudo apt --purge remove libcudnn7 libcudnn7-dev libnccl2
sudo apt-get autoremove

安装 Nvidia Driver,通过apt在standard Ubuntu Repository进行driver的安装：

1. 参考博文2中，使用图形化安装，我本地出现了无法应用修改，故采用命令行安装。

ubuntu-drivers devices
sudo apt install nvidia-driver-470  #此处为的recommnened为470

安装成功后，重启电脑

reboot

查看

nvidia-smi

根据推荐安装11.4 去CUDA Toolkit Archive | NVIDIA Developer

Cuda Toolkit可以通过两种方式安装，一种是distribution-specific packages (即类似与Debian packages), 另一种是distribution-independent package (即runfile packages)。
本文采用了Debian的方式（使用apt）进行了Nvidia driver的安装，因此CUDA也继续采用Debian方式进行安装

安装完成后，输入nvcc --version不显示所安装CUDA的版本号，需要添加以下路径到~/.bashrc(root隐藏文件 ls-a) ：

export LD_LIBRARY_PATH=/usr/local/cuda-xx.xx/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-xx.xx/bin:$PATH

nvcc --version

 Pytorch11.4

pytorch11.4 也可安装 11.3，此处pip3安装11.3，参考博文2

注意pip3必须20.0++以上，若不是，先升级pip3

pip3 install --upgrade pip