1. 程式人生 > >Mac下使用pip安裝Pytorch

Mac下使用pip安裝Pytorch

由於Mac不支援CUDA,所以使用下面的命令即可安裝pytroch,分兩種python2.7版和python3.5及以上。

Python 2.7:

sudo pip install torch torchvision
Python 3.5 & Python 3.6

sudo pip3 install torch torchvision

Pytorch官網:https://pytorch.org/

如果想安裝其它版本的Pytorch,可進入Pytorch官網根據自己的配置的情況選擇,官網會給出安裝的具體命令。