1. 程式人生 > >pytorch 0.4.1+ubuntu18.04 +python3.67+cuda9.2 安裝

pytorch 0.4.1+ubuntu18.04 +python3.67+cuda9.2 安裝

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