1、PyTorch pip安装
个人推荐使用pip的方式进行安装,这也是官方所推荐的标准方式,如下所示:
https://pytorch.org/get-started/locally/
2、PyTorch whl安装
当我们需要安装特定CUDA版本的PytTorch的时候,掌握whl文件的安装则是十分必要。
PyTorch whl文件的下载地址是:https://download.pytorch.org/whl/torch_stable.html
2.1、PyTorch版本与Python版本的关系
安装PyTorch要注意对应Python的版本,如下所示:
更多版本介绍,可以查看这里:https://github.com/pytorch/vision#installation