ubuntu20.04+1065显卡+460nvidia-drivers+cuda11.2+pytorch安装

1.Ubuntu 安装

这一步最简单， 就不多说了， 就是去官网下载镜像， 并且解压到U盘进行安装即可

2. 显卡驱动安装

输入这个命令，会列出最推荐的驱动，下面这个是对应的460

ubuntu-drivers devices

== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001C04sv00001B4Csd0000129Cbc03sc00i00
vendor   : NVIDIA Corporation
model    : GP106 [GeForce GTX 1060 5GB]
driver   : nvidia-driver-460-server - distro non-free
driver   : nvidia-driver-450-server - distro non-free
driver   : nvidia-driver-418-server - distro non-free
driver   : nvidia-driver-450 - distro non-free
driver   : nvidia-driver-390 - distro non-free
driver   : nvidia-driver-460 - distro non-free recommended
driver   : xserver-xorg-video-nouveau - distro free builtin

现在可以直接输入

sudo apt-get install nvidia-460

或者输入下面这个也行

sudo ubuntu-drivers autoinstall

执行完之后输入：

nvidia-smi

就可以查看显卡的详细信息

3. 安装CUDA11.2

调整gc++版本

sudo apt-get install gcc-7 g++-7

然后看一下目前使用的是GCC版本， 我的安装后只有一个7

sudo update-alternatives --display gcc

如果已经有9了，需要把7设置为优先

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 9
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 1
sudo update-alternatives --display gcc
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-7 9
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-9 1
sudo update-alternatives --display g++

接下来下载cuda
https://developer.nvidia.com/cuda-toolkit-archive
我直接安装的最新版11.2，我这里下载和安装的命令如下

wget https://developer.download.nvidia.com/compute/cuda/11.2.1/local_installers/cuda_11.2.1_460.32.03_linux.run
sudo sh cuda_11.2.1_460.32.03_linux.run

执行的时候会卡住， 建议多等等， 如果不行， 就重启了再执行

添加环境变量

gedit ~/.bashrc

这里的cuda版本按照你自己的相应修改

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

设置生效

source ~/.bashrc

到这里执行

nvcc -V

如果没有反映，则重启一下试试

4. 安装pytorch

到pytorch官网，按照自己的配置选择直接安装就行了
这里我使用的是

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

我这里没有11.2, 因为测试不能连接GPU

安装完毕之后，打开python，输入

>>> import torch
>>> torch.cuda.is_available()
True
>>> 

这里显示true， 则代表大功告成了！

引用：
https://zhuanlan.zhihu.com/p/59618999
https://blog.csdn.net/Nemo_Great/article/details/108414540