天天看点

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