pytorch 0.4.1+ubuntu18.04 +python3.67+cuda9.2 安裝
阿新 • • 發佈:2018-11-09
pytorch 0.4.1+ubuntu18.04 +python3.67+cuda9.2 安裝
點選進入官網 https://pytorch.org/
方法一:pip安裝
Python3.6+pip安裝cpu版本
pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-linux_x86_64.whl pip3 install torchvision
Python3.6+pip安裝gpu版本
目前gpu版本支援cuda8.0,cuda9.0和cuda9.2,請選擇對應的版本下載安裝,不要同時執行下面三個命令!
pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-linux_x86_64.whl pip3 install torchvision
(我選的是這個)
方法二:Conda安裝
如果你是Anaconda|Python使用者,cpu版本 執行命令:
conda install pytorch-cpu torchvision-cpu -c pytorch
就可以完成安裝。
,如果你需要安裝cuda8.0、cuda9.0或cuda9.2的gpu版本,請執行:
conda install pytorch torchvision cuda92 -c pytorch
pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-win_amd64.whl
測試安裝是否成功
python命令下輸入:
import torch
print(torch.__version__)
如果不報錯,就是安裝成功!
如果輸出0.4.1,那麼恭喜Windows下的PyTorch0.4.1安裝成功!
最後需要安裝 torchvision:
pip3 install torchvision
pytorch 0.4.1+win10 +python3.66+cuda9.2 安裝
win10版:https://blog.csdn.net/qq_41895190/article/details/82761162