【问题标题】:error installing pytorch using pip on windows 10在 Windows 10 上使用 pip 安装 pytorch 时出错
【发布时间】:2020-03-19 15:53:57
【问题描述】:

我正在尝试使用 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\

显然,我做错了什么。

请帮忙!

【问题讨论】:

    标签: python python-3.x pip pytorch


    【解决方案1】:

    如果您没有 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
    

    对于 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

    【讨论】:

      【解决方案2】:

      我遇到了同样的问题。现在问题已解决。 (2020-05-31)

      1. 访问了 pytorch.org 网站
      2. 并在 pytorch.org 的主页上找到“本地快速启动”。 (它可以通过向下滚动找到)
      3. 检查系统的环境形式 (例如:Windows、pip、python、、、) 然后,您可以看到安装命令。 “点安装火炬===....”

      4. 复制安装命令

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

      祝你好运!!

      【讨论】:

        【解决方案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

        【讨论】:

          猜你喜欢
          • 2020-03-01
          • 2018-07-18
          • 2019-12-21
          • 2018-10-14
          • 1970-01-01
          • 1970-01-01
          • 2020-07-19
          • 2012-12-22
          • 2020-12-31
          相关资源
          最近更新 更多