安装环境
- Windows
- Anaconda
- Cuda+cuDNN
Linux和Windows操作系统的安装存在差异，步骤会有所不同，本教程主要针对Windows系统进行示例。
Anaconda集成了许多方便的包和工具，使用会更加方便，特别适合科学计算，深度学习的数据处理。
Cuda是英伟达显卡用于科学计算的框架，cuDNN是NVIDIA提供的用于深度神经网络加速的GPU加速库。
安装步骤
-
安装当前显卡合适的CUDA版本，下载cuDNN扩展包复制到CUDA安装目录里面的对应子目录，网上教程很多，此处省略。
-
安装Anaconda
-
进入pytorch官网，获取安装合适的安装命令，打开Anaconda Prompt执行安装，比如我的指令为：
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
-
运行以下程序试试看
import torch
torch.cuda.is_available() # 查看GPU是否可用，输出True则代表可用，false不可用
