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/
此時運作指令
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