1. 程式人生 > >ubuntu 安裝 pytorch (ubuntu16.04+pytorch0.4.1+python2.7+cuda8.0)

ubuntu 安裝 pytorch (ubuntu16.04+pytorch0.4.1+python2.7+cuda8.0)

這裡使用pip安裝pytorch,我試過cuda安裝,和電腦的配置沒相容,沒有安裝成功,後來發現使用pip安裝很簡單方便,就是用pip安裝
首先進入pytorch官網,往下拉會看到不同的安裝選項
根據自己電腦的配置選擇不同的選項,例如我選擇的stable版本、Linux系統、pip安裝、python版本為2.7、cuda為8.0,如下圖所示:
在這裡插入圖片描述
選擇完之後下面會出現安裝所需的命令,如下圖
在這裡插入圖片描述
先後執行前兩個命令就可以,網速慢的話只能慢慢等了,安裝成功第一條命令後是這樣的
在這裡插入圖片描述
安裝成功第二條命令後是這樣的
在這裡插入圖片描述
下面測試是否安裝成功
命令列中輸入

python
import torch
torch.__version__

沒有問題的話會輸出版本為0.4.1,如下圖所示
在這裡插入圖片描述