1、PyTorch whl文件安装

当我们需要安装特定CUDA版本的PytTorch的时候,掌握whl文件的安装则是十分必要。

PyTorch whl文件的下载地址是:https://download.pytorch.org/whl/torch_stable.html

1.1、PyTorch版本与Python版本的关系

安装PyTorch要注意对应Python的版本,如下所示:

pytorch1.png

更多版本介绍,可以查看这里:https://github.com/pytorch/vision#installation

1.2、PyTorch的版本与CUDA的版本关系

pytorch2.png