天天看點

pytorch安裝(cpu版本)

pytorch安裝(cpu版本)

    • 本地安裝
    • 線上安裝
    • 替換源

本地安裝&線上安裝(pip)

系統:Ubuntu 18.04

本地安裝

線上上安裝不行的情況下,轉本地安裝,但首先要搞清楚你要安裝的是哪個版本

首先在終端輸入

ls -l /usr/bin | grep python
           

檢視你系統已有的python版本,一般來說系統會自帶一個2.7和一個3.x的版本,因為适合我安裝的pytorch版本适用python版本隻有3.5以上的。而系統預設的python一般是2.7,檢視自己系統預設python版本的方法:

python --version
           

我的預設版本是3.6,這個時候就要把python的版本切換到3.x版本,具體方法可以看我的另一篇博文切換系統python版本

在官網上能看到你要安裝的pytorch版本,在以下網址找到相應的whl檔案

最全的當然是在pytorch自己為下載下傳網站上,但速度會很慢,也可以去國内的一些鏡像源找一找

pytorch官方下載下傳位址

https://download.pytorch.org/whl/torch_stable.html

清華鏡像源

https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

下載下傳下來之後終端cd進入到所下載下傳檔案的檔案夾

sudo pip3 install torchpkg-name.whl
sudo pip3 install torchvisionpkg-name.whl
           

這裡的torchpkg-name.whl、torchvisionpkg-name.whl要換成你自己下載下傳的檔案名

之後連上國内源再下載下傳一個numpy的包(是一個pytorch使用必要的包)

sudo pip3 install numpy -i https://pypi.mirrors.ustc.edu.cn/simple/
           

到這裡就安裝完成了

進入終端驗證

輸入python進入consle

>>>import torch
>>> import torchvision
>>>print(torch.__version__)
           

不報錯即安裝成功

如果torchvision報錯嘗試指令

pip install --no-deps torchvision
           

線上安裝

由于在使用pip安裝時預設從國外的源下載下傳,是以為了保證下載下傳速度也為了防止中途由于網絡不好導緻下載下傳中斷,一般要進行源替換(見第三點)

首先去pytorch的官網檢視安裝指令,雖然很難安裝上,但他會提示你要安裝的版本

pytorch官網

https://pytorch.org/

pytorch安裝(cpu版本)

此時運作指令

pip install torch==1.5.1+cpu torchvision==0.6.1+cpu -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.mirrors.ustc.edu.cn/simple/
           

不出意外便能順利安裝,但如果标紅報錯就換本地安裝吧

我在直接輸入官網給出的指令後出現了"could not find a version that satisfies the requirement torch"的報錯,然後我在臨時換源之後順利進行,但安裝到一半就冒紅了,再次重新安裝程序就會被殺死,之後我就換到了本地的方式來安裝

(pip3安裝同理)

替換源

pip臨時使用國内源

pip install 庫名 -i 鏡像位址

執行個體:
pip install numpy -i https://pypi.mirrors.ustc.edu.cn/simple/
           

常用鏡像源

// 中國科技大學
https://pypi.mirrors.ustc.edu.cn/simple/

// 豆瓣
http://pypi.douban.com/simple/

// 清華大學
https://pypi.tuna.tsinghua.edu.cn/simple/

// 阿裡雲
http://mirrors.aliyun.com/pypi/simple/

// 中國科學技術大學
http://pypi.mirrors.ustc.edu.cn/simple/

// 華中理工大學
http://pypi.hustunique.com/

// 山東理工大學
http://pypi.sdutlinux.org/
           

永久修改

// pip更新到最新版本
pip install pip -U
// 更換國内源
pip config set global.index-url https://pypi.mirrors.ustc.edu.cn/simple/
           

非pip安裝其餘軟體安裝更換源的辦法

(也可以在“軟體與更新”中進行設定)

備份之前的源

cp /etc/apt/sources.list /etc/apt/sources.list.bak
           

删除原有的源更換

sudo gedit /etc/apt/sources.list
           

一些常用的國内源

阿裡雲

deb http://mirrors.aliyun.com/ubuntu/ xenial main
deb-src http://mirrors.aliyun.com/ubuntu/ xenial main

deb http://mirrors.aliyun.com/ubuntu/ xenial-updates main
deb-src http://mirrors.aliyun.com/ubuntu/ xenial-updates main

deb http://mirrors.aliyun.com/ubuntu/ xenial universe
deb-src http://mirrors.aliyun.com/ubuntu/ xenial universe
deb http://mirrors.aliyun.com/ubuntu/ xenial-updates universe
deb-src http://mirrors.aliyun.com/ubuntu/ xenial-updates universe

deb http://mirrors.aliyun.com/ubuntu/ xenial-security main
deb-src http://mirrors.aliyun.com/ubuntu/ xenial-security main
deb http://mirrors.aliyun.com/ubuntu/ xenial-security universe
deb-src http://mirrors.aliyun.com/ubuntu/ xenial-security universe
           

清華

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-security main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-security main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-proposed main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-proposed main restricted universe multiverse
           

中科大

deb https://mirrors.ustc.edu.cn/ubuntu/ focal main restricted universe multiverse
deb-src https://mirrors.ustc.edu.cn/ubuntu/ focal main restricted universe multiverse

deb https://mirrors.ustc.edu.cn/ubuntu/ focal-security main restricted universe multiverse
deb-src https://mirrors.ustc.edu.cn/ubuntu/ focal-security main restricted universe multiverse

deb https://mirrors.ustc.edu.cn/ubuntu/ focal-updates main restricted universe multiverse
deb-src https://mirrors.ustc.edu.cn/ubuntu/ focal-updates main restricted universe multiverse

deb https://mirrors.ustc.edu.cn/ubuntu/ focal-backports main restricted universe multiverse
deb-src https://mirrors.ustc.edu.cn/ubuntu/ focal-backports main restricted universe multiverse

## Not recommended
# deb https://mirrors.ustc.edu.cn/ubuntu/ focal-proposed main restricted universe multiverse
# deb-src https://mirrors.ustc.edu.cn/ubuntu/ focal-proposed main restricted universe multiverse

           

163

deb http://mirrors.163.com/ubuntu/ bionic main restricted universe multiverse
deb http://mirrors.163.com/ubuntu/ bionic-security main restricted universe multiverse
deb http://mirrors.163.com/ubuntu/ bionic-updates main restricted universe multiverse
deb http://mirrors.163.com/ubuntu/ bionic-proposed main restricted universe multiverse
deb http://mirrors.163.com/ubuntu/ bionic-backports main restricted universe multiverse
deb-src http://mirrors.163.com/ubuntu/ bionic main restricted universe multiverse
deb-src http://mirrors.163.com/ubuntu/ bionic-security main restricted universe multiverse
deb-src http://mirrors.163.com/ubuntu/ bionic-updates main restricted universe multiverse
deb-src http://mirrors.163.com/ubuntu/ bionic-proposed main restricted universe multiverse
deb-src http://mirrors.163.com/ubuntu/ bionic-backports main restricted universe multiverse