如何在Windows上安装PyTorch?

13

我正在尝试在Windows上安装PyTorch,已有可用版本但出现错误。

conda install -c peterjc123 pytorch=0.1.12

7
错误是什么? - Nico Haase
4
注意:这里没有依赖于本答案的必要性,您可以在 https://pytorch.org 找到安装指南。 - AMC
是的,它帮助我在官方网站上设置所需的设置,并复制适当的命令来安装PyTorch。 - LordDraagon
16个回答

31

警告:除非您有非常特殊的原因不这样做,否则请遵循来自https://pytorch.org的官方安装说明。它们更有可能准确且最新。


以下是如何在Windows上使用Anaconda从官方渠道安装PyTorch软件包的步骤,截至本评论撰写时(2020年3月31日):

没有CUDA的PyTorch:

conda install pytorch torchvision cpuonly -c pytorch

使用CUDA 10.1的PyTorch:

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

2
为什么要安装没有CUDA的版本? - Lucas
5
如果他没有GPU怎么办? - Jeru Luke
3
不使用笨重的“conda”框架,能否进行安装? - PhilMacKay
1
@PhilMacKay 不能不安装那个笨重的“conda”框架吗?Conda并不是特别沉重,我想你指的是Anaconda? - AMC
@AMC 是的,我经常忘记我们可以在没有Anaconda的情况下安装“conda” ;) 不过,“conda”需要很长时间来解决软件包依赖关系。从PyTorch的网站上,我们可以使用 pip install [...] -f https://download.pytorch.org/whl/torch_stable.html。如果在PyPI上有兼容Windows的版本就太好了! - PhilMacKay
显示剩余2条评论

21

访问官网:http://pytorch.org/

  1. 选择操作系统为Windows
  2. 选择包管理器,例如pip或conda
  3. 选择你的Python版本
  4. 选择CUDA或选择none。您将根据所选项获得安装pytorch的命令。

例如,如果您在上述步骤中选择了Windows、pip、Python 3.6none,则将获取以下命令:

pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp36-cp36m-win_amd64.whl 
pip3 install torchvision

10

实际答案::


最好的方法是在官方网站上查看最新选项。以下为截至2020年3月仍能使用的选项:

# for windows 10, CUDA 10.1
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

#for windows 10, CUDA 9
conda install pytorch torchvision cudatoolkit=9.2 -c pytorch -c defaults -c numba/label/dev 

之前的回答已过时。


看起来作者(peterjc123)两天前发布了conda包,以在Windows上安装PyTorch 0.3.0。以下是一个副本:

# for Windows 10 and Windows Server 2016, CUDA 8
conda install -c peterjc123 pytorch cuda80

# for Windows 10 and Windows Server 2016, CUDA 9
conda install -c peterjc123 pytorch cuda90

# for Windows 7/8/8.1 and Windows Server 2008/2012, CUDA 8
conda install -c peterjc123 pytorch_legacy cuda80

来源:https://github.com/pytorch/pytorch/issues/494#issuecomment-350527200


这个已经过时了。 - AMC
感谢您更新您的答案! :) - AMC

10

如果您正在尝试在Windows 10上安装,并且没有安装Anaconda,则以下是最佳选项:

Python 2.7

pip install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp27-cp27mu-linux_x86_64.whl
pip install torchvision
如果上述命令无法运行,则您使用的是Python 2.7 UCS2版本,请使用此命令。
pip install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp27-cp27m-linux_x86_64.whl

Python 3.5

pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision

Python 3.6

pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision

Python 3.7

pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp37-cp37m-win_amd64.whl
pip3 install torchvision

你没有安装Anaconda,这是什么意思? - AMC
这意味着您的计算机上没有安装Anaconda发行版。 - Navin Dalal

4
更新于2019年6月:现在PyTorch有专门的conda渠道,可以使用anaconda轻松安装。在执行pytorch生成的命令之前,该命令需要一些依赖项才能成功运行。例如,我选择了稳定的pytorch 1.1版本,Python 3.6和Cuda 10.0构建。Pytorch页面生成的命令如下所示:
``` conda install pytorch torchvision cudatoolkit=10.0 -c pytorch ```
但如果像我一样创建了一个新的conda环境,则无法正常运行。设置pytorch的逐步过程如下:
第一步,安装cudatoolkit,命令如下:
``` conda install -c anaconda cudatoolkit=10.0 ```
第二步,安装mkl_fft,命令如下:
``` conda install -c anaconda mkl_fft ```
接下来假设没有出现更多的依赖问题。使用以下命令设置pytorch:
``` conda install -c pytorch pytorch ```
这对我起作用了。但在开始pytorch设置之前,我已经设置了我的新conda环境,包括scikit-learn和jupyter笔记本电脑。因此,如果出现任何依赖性问题,最好也安装scikit-learn和jupyter notebook。

但是如果像我一样创建了一个新的conda环境,它将无法工作。你为什么需要手动安装这些依赖项? - AMC

3

你好,你能帮忙在Win32位系统上安装PyTorch吗?他们说不支持Win32位系统... - ERJAN
真正的答案。 - AMC
2
我认为你应该用红色圈出“运行此命令”。我之前看了一个链接或者更明显的东西,但是这个命令直接在我的面前哈哈。 - Sauleil

1

1
你可能想考虑使用Docker for Windows。这将使您能够像在Linux上一样安装pytorch。尽管如此,我认为DfW的CUDA支持有限,因此如果您打算使用CUDA,则可能需要探索其他选项。

1

尝试运行:

conda install -c pytorch pytorch

该命令将更新/安装:conda、cudatoolkit和pytorch。

0

为什么要改变目录? - AMC
因为conda可能不在路径中。 - Sergey Batsura

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接