原文鏈接

windows下使用pip安裝torch模塊，出現(xiàn)錯誤：

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/u5uIjnABGXTJGo4Z3_dZQw

指定安裝包的方法，指定安裝包進(jìn)行安裝。

方法二

創(chuàng)建一個環(huán)境

打開Anaconda prompt，輸入：

conda create -n myroot python=3.5

myroot為自己定義的環(huán)境名稱，對應(yīng)的python3.5版本。

然后輸入：

activate myroot

激活環(huán)境。

通過：

python --version
nvcc --version

分別查看python和cuda版本。

安裝 pytorch

根據(jù)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

安裝成功:

學(xué)習(xí)更多編程知識，請關(guān)注我的公眾號：

[代碼的路]