PyTorch快速上手教程

Pytorch安装

安装CUDA：

1. 首先安装英伟达显卡驱动（如果已安装则跳过）: https://www.nvidia.com/Download/index.aspx
2. 检查显卡支持的CUDA版本
   • 查看本地显卡支持的最高CUDA版本：nvidia-smi
   • Nvidia官网查看：GPU型号与CUDA版本对应列表
   • 查看PyTorch支持的CUDA版本：PyTorch 官方安装教程
3. 下载安装CUDA
   • 最新版本：CUDA下载
   • 历史版本：CUDA下载
4. 安装cuDNN
   • cuDNN 下载
   • 下载好zip包后解压，将子目录复制到cuda安装根目录
   • 将cuDNN子目录(bin、include、lib)都添加到系统环境变量Path中
5. 添加环境变量
   • 在系统环境变量Path中添加
6. 检查安装结果
   • nvcc -V

安装PyTorch：

1. 使用pip安装对应cuda版本的：PyTorch 官方安装教程
2. 验证安装

   import torch
   torch.__version__  # 查看torch版本
   torch.cuda.is_available()  # 查看cuda能否使用，也就是GPU能否使用
   torch.version.cuda  # 编译pytorch时使用的cuda版本
   torch.cuda.get_device_name(0)  # NVIDIA 5080 等