pytorch 安装

1、pip install torch安装时默认是使用国外的源。如果是在国内，而且包比较大（2G），下载可能会很慢。因此，直接下载对应版本的 whl 包，再离线安装。

下载链接：https://download.pytorch.org/whl/torch_stable.html

安装命令：conda install <whl file path>或pip install <whl file path>

2、下载 whl 包时，要注意本地机器是否有 nvidia 显卡，如果没有只需下载 cpu 版本的包，如果有可以下载 cuda 版本的包。

3、通常 pytorch 的安装要包含 torch，torchvision，torchaudio 三件套，要注意的三件套的版本要对应匹配。如果不清楚版本对应关系，可以先下载 torch 的 whl 包，完成安装。

由于 torchvision 和 torchaudio 包本身比较小，可以直接用安装命令：pip install torchvision torchaudio 或 conda install torchvision torchaudio 完成安装。此时会默认下载与已安装 torch 的对应版本的 torchvision 和 torchaudio.