安装环境

• Windows
• Anaconda
• Cuda+cuDNN

Linux和Windows操作系统的安装存在差异，步骤会有所不同，本教程主要针对Windows系统进行示例。

Anaconda集成了许多方便的包和工具，使用会更加方便，特别适合科学计算，深度学习的数据处理。

Cuda是英伟达显卡用于科学计算的框架，cuDNN是NVIDIA提供的用于深度神经网络加速的GPU加速库。

安装步骤

1. 安装当前显卡合适的CUDA版本，下载cuDNN扩展包复制到CUDA安装目录里面的对应子目录，网上教程很多，此处省略。

2. 安装Anaconda

3. 进入pytorch官网，获取安装合适的安装命令，打开Anaconda Prompt执行安装，比如我的指令为：conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
   

4. 运行以下程序试试看

import torch
torch.cuda.is_available() # 查看GPU是否可用，输出True则代表可用，false不可用