Pytorch 安装-Ubuntu 20.04

1. 安装显卡驱动，ubuntu-drivers devices，找到recommended驱动版本

2. sudo apt install安装指定版本驱动

3. nivida-smi检查是否安装成功

4. CUDA Toolkit 11.7 Update 1 Downloads | NVIDIA DeveloperResources CUDA Documentation/Release NotesMacOS Tools Training Sample Code Forums Archive of Previous CUDA Releases FAQ Open Source PackagesSubmit a BugTarball and Zip Archive Deliverableshttps://developer.nvidia.com/cuda-downloads安装cuda的runfile文件。ps：pytorch可以不安装此步骤，可略过

5. 安装cudnn。ps：pytorch可略过

6. 安装anaconda

7. 安装教育版pycharm

8. 安装pip。更新pip源

9. pycharm创建虚拟环境

10. 使用pip方式安装pytorch（自带cuda和cudnn）

Start Locally | PyTorchAn open source machine learning framework that accelerates the path from research prototyping to production deployment.https://pytorch.org/get-started/locally/

11. 验证：

python

import torch, torchvision

print(torch.__version__)

print(torch.cuda.is_available())

print(torch.version.cuda)

from torch.backends import cudnn

print(cudnn.is_available())

print(torch.backends.cudnn.version())