Anaconda 環境下Pytorch
阿新 • • 發佈:2018-12-16
目前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