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