Pytorch(一)环境的安装

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qq_36022260/article/details/83211087

目前python2.7不支持CONDA安装pytorch，请安装Python 3再安装Pytorch

如果你是Python 3.5环境,你可以根据你的CUDA情况进行选择安装

# cuda8 conda install pytorch -c pytorch 

# cuda9.0 conda install pytorch cuda90 -c pytorch 

# cuda9.1 conda install pytorch cuda91 -c pytorch 

# cudanone conda install pytorch-cpu -c pytorch

如果你是python3.6环境，你可以参考下面的方法
 

# cuda8 conda install pytorch -c pytorch

# cuda9.0 conda install pytorch cuda90 -c pytorch 

# cuda9.1 conda install pytorch cuda91 -c pytorch 

# cudanone conda install pytorch-cpu -c pytorch

Windos使用PIP安装pythorch

目前python2.7不支持PIP安装pytorch，请安装Python 3再安装Pytorch

如果你是Python 3.5环境,你可以根据你的CUDA情况进行选择安装

# 没有cuda的情况
pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp35-cp35m-win_amd64.whl

# cuda8环境
pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-win_amd64.whl

# cuda9.0环境
pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp35-cp35m-win_amd64.whl

# cuda9.1环境
pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp35-cp35m-win_amd64.whl

如果你是python3.6环境，你可以参考下面的方法：

# 没有cuda的情况
pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp36-cp36m-win_amd64.whl

# cuda8环境
pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp36-cp36m-win_amd64.whl

# cuda9.0环境
pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp36-cp36m-win_amd64.whl

# cuda9.1环境
pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl

Windos安装torchvision

不管你Windos下是conda还是PIP安装的Pytorch 0.4.0，都需要使用PIP来安装torchvision

# 最后安装torchvision
pip3 install torchvision

原文出处： https://ptorch.com/news/164.html