安装 anaconda

anaconda是一个开源的Python发行版本，其包含了conda、Python等180多个科学包及其依赖项。

官网地址：https://www.anaconda.com/products/distribution#windows

下载安装，验证是否安装成功

$ conda --version
conda 4.12.0

// 依赖列表
$ conda list

安装Pytorch

官网地址：https://pytorch.org/

这里的计算平台可以选择简单一些入门，因为我的是A卡，所以直接选择CPU为计算平台的安装，如果是N卡，可以选择CUDA计算平台

CUDA下载地址：https://developer.nvidia.com/cuda-downloads

两种方式均可

$ conda install pytorch torchvision torchaudio cpuonly -c pytorch

$ pip3 install torch torchvision torchaudio

pip和pip3都是python的依赖管理工具，只是版本不同

测试torch

import torch

# torch version
print(torch.__version__)
# is support gpu
print('gpu:',torch.cuda.is_available())

1.11.0+cpu
gpu: False

打开文档

jupyter notebook