Ubuntu 18.04 安装显卡驱动+PyTorch+Cuda+Cudnn

1. 安装显卡驱动

Step 1

ubuntu-drivers devices

Step 2

sudo ubuntu-drivers autoinstall

等待安装完毕后reboot

2. 安装anaconda

Step 1
在conda官网上下载anaconda

Step 2
在下载路径下安装anaconda（注意不要使用sudo！）

Step 3
安装完成后重新打开一个terminal

Step 4
创建一个conda的新环境

conda create -n env_name

Note：不要写Python

3. 安装PyTorch+Cuda+Cudnn

在PyTorch官网复制链接

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

静等下载、安装完毕即可（没有代理就换源）