在Windows 10上使用pip安装PyTorch时出现错误

Question

我正在尝试使用pip安装pytorch

pip install torch

或者

pip3 install torch===1.3.1 torchvision===0.4.2 -f https://download.pytorch.org/whl/torch_stable.html

使用Python 3.7.4

和Python 3.8（最新的稳定版本）

都支持32位和64位。

并获取到以下信息：

Collecting torch   Using cached
https://files.pythonhosted.org/packages/f8/02/880b468bd382dc79896eaecbeb8ce95e9c4b99a24902874a2cef0b562cea/torch-0.1.2.post2.tar.gz
Collecting pyyaml (from torch)   Downloading
https://files.pythonhosted.org/packages/bc/3f/4f733cd0b1b675f34beb290d465a65e0f06b492c00b111d1b75125062de1/PyYAML-5.1.2-cp37-cp37m-win_amd64.whl
(215kB)
    100% |████████████████████████████████| 225kB 1.2MB/s Installing collected packages: pyyaml, torch   Running setup.py install for torch
... error
    Complete output from command C:\Noam\Code\threadart\stav-rl\venv\Scripts\python.exe -u -c "import
setuptools,
tokenize;__file__='C:\\Users\\noams\\AppData\\Local\\Temp\\pip-install-djc6s2t8\\torch\\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\noams\AppData\Local\Temp\pip-record-zohv2zo7\install-record.txt
--single-version-externally-managed --compile --install-headers C:\Noam\Code\threadart\stav-rl\venv\inclu de\site\python3.7\torch:
    running install
    running build_deps
    Traceback (most recent call last):
      File "<string>", line 1, in <module>
      File "C:\Users\noams\AppData\Local\Temp\pip-install-djc6s2t8\torch\setup.py",
line 265, in <module>
        description="Tensors and Dynamic neural networks in Python with strong GPU acceleration",
      File "C:\Noam\Code\threadart\stav-rl\venv\lib\site-packages\setuptools-40.8.0-py3.7.egg\setuptools\__init__.py",
line 145, in setup
      File "C:\Python37_x64\lib\distutils\core.py", line 148, in setup
        dist.run_commands()
      File "C:\Python37_x64\lib\distutils\dist.py", line 966, in run_commands
        self.run_command(cmd)
      File "C:\Python37_x64\lib\distutils\dist.py", line 985, in run_command
        cmd_obj.run()
      File "C:\Users\noams\AppData\Local\Temp\pip-install-djc6s2t8\torch\setup.py",
line 99, in run
        self.run_command('build_deps')
      File "C:\Python37_x64\lib\distutils\cmd.py", line 313, in run_command
        self.distribution.run_command(command)
      File "C:\Python37_x64\lib\distutils\dist.py", line 985, in run_command
        cmd_obj.run()
      File "C:\Users\noams\AppData\Local\Temp\pip-install-djc6s2t8\torch\setup.py",
line 51, in run
        from tools.nnwrap import generate_wrappers as generate_nn_wrappers
    ModuleNotFoundError: No module named 'tools.nnwrap'

    ---------------------------------------- Command "C:\Noam\Code\threadart\stav-rl\venv\Scripts\python.exe -u -c "import
setuptools,
tokenize;__file__='C:\\Users\\noams\\AppData\\Local\\Temp\\pip-install-djc6s2t8\\torch\\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\noams\AppData\Local\Temp\pip-record-zohv2zo7\install-record.txt
--single-version-externally-managed --compile --install-headers C:\Noam\Code\threadart\stav-rl\venv\include\site\python3.7\torch" 
failed with error code 1 in
C:\Users\noams\AppData\Local\Temp\pip-install-djc6s2t8\torch\

显然，我做错了什么。

请帮帮我！

- Gulzar

Answer 1

我曾经有过同样的问题，现在这个问题已经解决了。（2020-05-31）

1. 访问 pytorch.org 网站。
2. 在 pytorch.org 的首页上找到 "QUICK START LOCALLY"（可以通过向下滚动找到）。

3. 检查您的系统环境形式 （例如：Windows、pip、python等） 然后，您可以看到安装命令。 "pip install torch===.... "

4. 复制安装命令

5. 并在您的系统上执行该命令。

祝您好运！！

- WangSung

Answer 2

如果您没有GPU（仅适用于CPU），请使用以下命令：

• 版本1.2： conda install pytorch==1.2.0 torchvision==0.4.0 cpuonly -c pytorch

• 新版本： conda install pytorch torchvision cpuonly -c pytorch

或者

如果您有GPU，请使用以下命令（使用您自己的CUDA版本）：

• 版本1.2： conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

• 新版本： conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

建议使用Anaconda命令提示符安装PyTorch，如果可能的话，请使用Python 3.6，更加适合。Python 3.6对于使用开源库更加稳定。此外，您可以在下面链接中找到安装过程，如果需要的话， https://medium.com/@bryant.kou/how-to-install-pytorch-on-windows-step-by-step-cc4d004adb2a

如果您只想使用pip进行设置，则可以尝试如下：

对于Python 3.6：

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

For Python 3.7:

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

我认为在安装开源库时，Conda是比pip更好的选择。

- Joy

Answer 3

如果您在Windows上使用Anaconda（3.7），则需要通过.whl文件安装。

从这里下载.whl文件：- https://download.pytorch.org/whl/torch_stable.html 然后将其放入工作目录并使用pip进行安装，对于我的系统，我已将Anaconda的pip重命名为pip37。

pip37 install torch-1.5.1+cpu-cp37-cp37m-win_amd64.whl
pip37 install torchvision-0.6.1+cpu-cp37-cp37m-win_amd64.whl

它将在您的Anaconda环境中安装Torch 1.5和Torchvision 0.6.1。

- Kushal Bhavsar