原文链接
windows 下应用 pip 装置 torch 模块,呈现谬误:
ERROR: Command “python setup.py egg_info” failed with error code 1 in C:\Users\xxx\AppData\Local\Temp\pip-install-yqzlud5w\torch\
办法一
尝试参考文章:https://mp.weixin.qq.com/s/u5…
指定安装包的办法,指定安装包进行装置。
办法二
创立一个环境
关上 Anaconda prompt,输出:
conda create -n myroot python=3.5
myroot 为本人定义的环境名称,对应的 python3.5 版本。
而后输出:
activate myroot
激活环境。
通过:
python --version
nvcc --version
别离查看 python 和 cuda 版本。
装置 pytorch
依据 python 和 cuda 的版本抉择装置 GPU 或 CPU 版本:
# 应用 conda 装置
'conda,cuda8,python3.5': conda install pytorch -c pytorch
'conda,cuda9.0,python3.5': conda install pytorch cuda90 -c pytorch
'conda,cuda9.1,python3.5': conda install pytorch cuda91
'conda,cudanone,python3.5': conda install pytorch-cpu -c pytorch
'conda,cuda8,python3.6': conda install pytorch -c pytorch
'conda,cuda9.0,python3.6': conda install pytorch cuda90 -c pytorch
'conda,cuda9.1,python3.6': conda install pytorch cuda91 -c pytorch
'conda,cudanone,python3.6': conda install pytorch-cpu -c pytorch
#应用 pip 装置
'pip,cudanone,python3.5': pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp35-cp35m-win_amd64.whl
'pip,cuda8,python3.5': pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-win_amd64.whl
'pip,cuda9.0,python3.5': pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp35-cp35m-win_amd64.whl
'pip,cuda9.1,python3.5': pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp35-cp35m-win_amd64.whl
'pip,cudanone,python3.6': pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp36-cp36m-win_amd64.whl
'pip,cuda8,python3.6': pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp36-cp36m-win_amd64.whl
'pip,cuda9.0,python3.6': pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp36-cp36m-win_amd64.whl
'pip,cuda9.1,python3.6': pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl
装置胜利:
学习更多编程常识,请关注我的公众号:
代码的路