PyCharm 不会安装 PyTorch - 构建轮失败

Question

我已经尝试安装 PyTorch 和 TorchVision 一段时间来帮助我正在做的一些神经网络/机器学习项目。

我的问题是我似乎无法让安装工作，无论是在 PyCharm pip、Anaconda 提示符还是 cmd/使用 pip3 命令。

尝试通过 PyCharm 运行以下命令时得到的输出：

pip install --user pytorch==0.1.2

这是：

Collecting pytorch
  Using cached https://files.pythonhosted.org/packages/a9/41/4487bc23e3ac4d674943176f5aa309427b011e00607eb98899e9d951f67b/pytorch-0.1.2.tar.gz
Building wheels for collected packages: pytorch
  Running setup.py bdist_wheel for pytorch: started
  Running setup.py bdist_wheel for pytorch: finished with status 'error'
  Complete output from command C:\Users\Even\Anaconda3\envs\python.exe -u -c "import setuptools, tokenize;__file__='C:\\Users\\Even\\AppData\\Local\\Temp\\pycharm-packaging\\pytorch\\setup.py';f=getattr(tokenize, 'open', open)(__file__);code=f.read().replace('\r\n', '\n');f.close();exec(compile(code, __file__, 'exec'))" bdist_wheel -d C:\Users\Even\AppData\Local\Temp\pip-wheel-2sfgqmm7 --python-tag cp36:
  Traceback (most recent call last):
    File "<string>", line 1, in <module>
    File "C:\Users\Even\AppData\Local\Temp\pycharm-packaging\pytorch\setup.py", line 17, in <module>
      raise Exception(message)
  Exception: You should install pytorch from http://pytorch.org

  ----------------------------------------
  Running setup.py clean for pytorch
Failed to build pytorch
Installing collected packages: pytorch
  Running setup.py install for pytorch: started
    Running setup.py install for pytorch: finished with status 'error'
    Complete output from command C:\Users\Even\Anaconda3\envs\python.exe -u -c "import setuptools, tokenize;__file__='C:\\Users\\Even\\AppData\\Local\\Temp\\pycharm-packaging\\pytorch\\setup.py';f=getattr(tokenize, 'open', open)(__file__);code=f.read().replace('\r\n', '\n');f.close();exec(compile(code, __file__, 'exec'))" install --record C:\Users\Even\AppData\Local\Temp\pip-record-1jx5e1ke\install-record.txt --single-version-externally-managed --compile --user --prefix=:
    Traceback (most recent call last):
      File "<string>", line 1, in <module>
      File "C:\Users\Even\AppData\Local\Temp\pycharm-packaging\pytorch\setup.py", line 13, in <module>
        raise Exception(message)
    Exception: You should install pytorch from http://pytorch.org

    ----------------------------------------

  Failed building wheel for pytorch
Command "C:\Users\Even\Anaconda3\envs\python.exe -u -c "import setuptools, tokenize;__file__='C:\\Users\\Even\\AppData\\Local\\Temp\\pycharm-packaging\\pytorch\\setup.py';f=getattr(tokenize, 'open', open)(__file__);code=f.read().replace('\r\n', '\n');f.close();exec(compile(code, __file__, 'exec'))" install --record C:\Users\Even\AppData\Local\Temp\pip-record-1jx5e1ke\install-record.txt --single-version-externally-managed --compile --user --prefix=" failed with error code 1 in C:\Users\Even\AppData\Local\Temp\pycharm-packaging\pytorch\

我也试过这个命令（CMD）：

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

在此输出中返回的内容：

torch-0.4.1-cp36-cp36m-win_amd64.whl is not a supported wheel on this 
platform.

我可能遗漏了一些重要信息，但我一直在谷歌搜索、搜索论坛并询问我认识的人。没有人有答案。我希望你们中的一些人可能遇到过同样的问题。

谢谢。

Answer 1

首先，在本地终端验证安装是否完成。从https://pytorch.org/查看特定于 Windows 的安装

完成此操作后，请确保在 pycharm 中选择正确的 Python 解释器。请参考链接：https://www.jetbrains.com/help/pycharm/installing-uninstalling-and-upgrading-packages.html 来管理特定 Python 解释器的包。

Answer 2

1. 我在https://pytorch.org/ 上解决了这个问题，如错误所示。并遭受

   torch-0.4.1-cp36-cp36m-win_amd64.whl 不支持此轮 平台

。也是。但我发现是我的 CUDA 版本导致了错误。我以为是 8/9，但实际上是 10。您可以使用pip --version查看 pip 和 python 版本。

1. 还有一点我们应该注意。 Pip 和 Anaconda 以不同的方式安装软件包。以前，Pytorch 是由 Anaconda 安装的，但我无法在 Anaconda Prompt (No module named torch) 中导入。我必须使用 pip 安装才能修复错误。

希望这会有所帮助。