1. 程式人生 > >Anaconda 環境下Pytorch

Anaconda 環境下Pytorch

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