Windos使用CONDA/PIP安装官方pythorch 0.4.0/torchvision

Song • 823 次浏览 • 0 个回复 • 2018年04月26日

Windos使用CONDA安装pythorch

目前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

原创文章,转载请注明 :Windos使用CONDA/PIP安装官方pythorch 0.4.0/torchvision - pytorch中文网
原文出处: https://ptorch.com/news/164.html
问题交流群 :168117787
提交评论
要回复文章请先登录注册
用户评论
  • 没有评论
Pytorch是什么?关于Pytorch! pytorch回归网络不能使用数据并行