Linux安装pytorch方法

2022-04-01 11:05:46 浏览数 (1)

1、Pytorch介绍

PyTorch是一个基于Torch的Python开源机器学习库,用于自然语言处理等应用程序它主要由Facebook的人工智能小组开发,不仅能够实现强大的GPU加速,同时还支持动态神经网络,这一点是现在很多主流框架如TensorFlow都不支持的。PyTorch提供了两个高级功能:

  1. 具有强大的GPU加速的张量计算(如Numpy);
  2. 包含自动求导系统的深度神经网络。

2、Anaconda环境搭建

Anaconda是一个用于科学计算的Python发行版,支持Linux、Mac和Window系统,提供了包管理与环境管理的功能,可以很方便地解决Python并存、切换,以及各种第三方包安装的问题。包含了conda、Python在内的超过180个科学包及其依赖项。

2.1下载anaconda

在Anaconda3官方下载对应的系统版本,本教程以64位linux_16.04系统为例。官网下载地址如下:https://www.anaconda.com/products/individual#Downloads

如果下载太慢,可以选择清华镜像下载:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

2.1安装anaconda

安装前请去先官网下载最新的,直接打开终端

cd ~/Downloads

bash Annaconda...xxx.sh(文件名,文件名可以用tab健自动补全)

yes

sudo gedit ~/.bashrc (这一步配置环境变量)

输入你的账户密码进入 文本 (为了修改环境变量)

复制 终端末尾显示的一行( export ……………../bin/…….)到文本,然后保存

新开一个终端输入python 测试是否Anaconda安装成功

2.2添加镜像源

因国外网站下载网路不稳定,需添加国内镜像源。

方法一

在打开anaconda虚拟环境(base)环境后,在终端中输入如下命令。以添加清华大学镜像源为例:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/

conda config --set show_channel_urls yes 运行完命令即可

方法二

此外,网上有诸多添加镜像源的方法,直接搜索Linux系统添加镜像源即可。

2.3创建虚拟环境并切换到该虚拟环境

启动终端,输入创建虚拟环境的命令(本教程以创建python3.6版本为例):

conda create --name torchtest python=3.6

如需安装其他版本或者建立其它虚拟环境范例命令如下:

conda create --name (此处替换为您创建的项目名) (此处对应您所需python版本)

创建虚拟环境如果非常慢,很可能是由于网络原因,网速慢导致,所以在此之前一定要修改镜像。中间会有提示,创建该环境需要安装什么,问你是y 还是n,输入y后回车,就会安装。

然后激活虚拟环境命令为:

代码语言:javascript复制

此处没有任何错误,即为成功激活torchtest虚拟环境。

可以进行验证,输入python 看看是否为你刚才安装的版本。

3.在创建的虚拟环境中安装pytorch

3.1命令安装pytorch

首先我们需要去pytorch官方网站选择我们所需安装的版本,官网网址:https://pytorch.org/get-started/locally/

选定好后,复制官网最下行Run this Command提供的命令以此图为例,我们运行提供的conda install pytorch torchvision torchaudio cpuonly -c pytorch

但是如果安装之前的版本在哪里找呢,我们以安装pytoch 1.0.0 cpu版本为例。如图:

然后找到pytorch1.0.0版本如图所示:

在我们的torchtest虚拟环境中运行此命令:conda install pytorch-cpu==1.0.0 torchvision-cpu==0.2.1 cpuonly -c pytorch

安装成功即可。

3.2手动安装pytorch

如果该命令行安装不上,可考虑下面的本地安装方式。pytorch 1.0 cpu版的本地安装方式:

【下载】在pytorch网站https://download.pytorch.org/whl/cpu/torch_stable.html

【移动】把包下载下来之后,放到命令行前的目录处,放到哪里都可以,但一定要cd到该目录下,我的浏览器自动下载到Download中,因此

cd Download

pip install torchxxxxxxxxxx.whl(文件名)

到此pytorch就安装成功了。

3.3测试pytorch是否安装成功

在torchtest虚拟环境中,首先输入python,然后import torch,如果没有任何报错,直接下一行即是安装成功,如图所示:

到此在window10系统下安装Anaconda、Pytorch就完成了。

4.附录

如果想安装其他的包例如pandas,matplotlib,scipy,jupyter,tqdm,visdom ,torchnet,jieba,numpy,fire,torchtext,ipdb,numpy就在torchtest虚拟环境下输入以下命令即可:

conda install(包的名称)

包的安装优选选择conda速度快。有些conda无法安装的包【可以是由于镜像中没有类似版本的包】,可以再尝试pip install 方式。

pip install(包的名称)

例如安装pandas为例:

conda install pandas

显示done或者successful即安装成功。如果conda不能按照安装利用pip安装即可 在torchtest中输入命令pip install pandas即可,如图所示:

检验是否安装成功,与pytorch检验过程相同。

没有任何报错即是安装成功。

如果有错误,可以搜索错误行,去百度搜索该错误,大多数能得到解决。

0 人点赞