撰写于    浏览:671 次

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的版本,如下所示:

pytorch1.png

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

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

pytorch2.png


如需评论,请填写表单。

提交前先勾选