在Linux系统中下载安装PyTorch及主要功能介绍

Facebook 最新人工智能开源框架 PyTorch 1.0 预览版本发布下载了，该版本搭载了很多新功能，同时我们介绍在 Linux 系统下安装 PyTorch 的方法。

PyTorch 简介

PyTorch 是一个基于 Python 的科学计算库，同时也是 Facebook 在人工智能项目中广泛使用的开源 AI 框架。PyTorch 利用 GPU 超强的运算能力来实现复杂的张量计算和深度神经网络。因此，它被世界各地的研究人员和开发人员广泛使用。

PyTorch 1.0 预览版本主要功能介绍

1、JIT

JIT 是一个编译工具集，使研究和生产更加接近。它包含一个基于 Python 语言的叫做 Torch Script 的脚本语言，也有能使现有代码与它自己兼容的方法。

2、全新的 torch.distributed 库: “C10D”

“C10D” 能够在不同的后端上启用异步操作，并在较慢的网络上提高性能。

3、 C++ 前端 (实验性功能)

虽然它被特别提到是一个不稳定的 API（估计是在预发行版中），这是一个 PyTorch 后端的纯 C++ 接口，遵循 API 和建立的 Python 前端的体系结构，以实现高性能、低延迟的研究和开发直接安装在硬件上的 C++ 应用程序。

PyTorch 下载地址

pytorch网站

pytorch GitHub主页

在 Linux 系统下安装 PyTorch 的方法

建议使用 conda 来安装 PyTorch v1.0rc0，开发人员建议使用 conda。

1.安装前提

Linux

Pip

Python

CUDA (对于使用 Nvidia GPU 的用户)

先安装 Pip，如果你还没有安装，请参考在Ubuntu 18.04系统中安装pip并使用pip命令一文。

请注意，PyTorch 具有 GPU 和仅限 CPU 的不同安装包。你应该安装一个适合你硬件的安装包。

2.安装 PyTorch 的旧版本和稳定版

如果你想在 GPU 机器上安装稳定版（0.4 版本），使用：

pip install torch torchvision

使用以下两个命令，来安装仅用于 CPU 的稳定版：

pip install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp27-cp27mu-linux_x86_64.whl

pip install torchvision

2.安装 PyTorch 1.0 候选版本

使用如下命令安装 PyTorch 1.0 RC GPU 版本：

pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cu92/torch_nightly.html

如果没有GPU，并且更喜欢使用 仅限 CPU 版本，使用如下命令：

pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html

在 Linux 系统下验证 PyTorch 安装

使用如下简单的命令，启动终端上的 python 控制台：

python

现在，按行输入下面的示例代码以验证您的安装：

from __future__ import print_function

import torch

x = torch.rand(5, 3)

print(x)

你应该得到如下输出：

tensor([[0.3380, 0.3845, 0.3217],

[0.8337, 0.9050, 0.2650],

[0.2979, 0.7141, 0.9069],

[0.1449, 0.1132, 0.1375],

[0.4675, 0.3947, 0.1426]])

若要检查是否可以使用 PyTorch 的 GPU 功能， 可以使用以下示例代码:

import torch

torch.cuda.is_available()

输出结果应该是：

True

结语

本文介绍了 PyTorch 1.0 预览版本主要功能及在 Linux 系统中的安装方法，目前有一些重量级应用是基于 PyTorch 的，比如 Detectron，它是 Facebook AI 研究院的软件系统，可以智能地进行对象检测和分类，它之前是基于 Caffe2 的。还有 Unsupervised Sentiment Discovery（一个广泛应用于社交媒体的一些算法）及 vid2vid（一个逼真的视频到视频的转换）。总之，PyTorch 是一个很有前景的项目，待 PyTorch 1.0 稳定版本发布后它将被更多的开发者所使用。

相关主题

Facebook致力于人工智能，将开源PyTorch 1.0 AI框架