fastai 系列教程(一)- 安装

PyTorch入门实战教程

前两天伴随着 PyTorch 开发者大会的召开,PyTorch 1.0 预览版也终于和大家见面。随之发布的还有 fastai 深度学习库,相当于 PyTorch 的 Keras。fastai 基于 PyTorch,提供简单易用的 API 接口,用更少的代码实现常用任务的模型搭建和训练。

我们 PyTorch 中文网为大家撰写和整理了一套 fastai 的快速入门教程,今天来讲 fastai 的安装。

系统需求

1. Python: 版本 3.6 或以上

2. 操作系统:

因为 fastai 1.0 是基于 PyTorch 1.0 的,所以我们必须要能够先安装 PyTorch 1.0。支持的安装方式:

系统GPUCPU
linuxbinarybinary
macsourcebinary
windowssourcesource

备注: binary = 直接可以安装, source = 必须从源码编译

通过 Conda 安装

通过下列两步即可安装,请严格按照先装第一步,再装第二步的顺序。

1. 安装每日编译 nightly 的 PyTorch,注意 cuda 的版本要和你自己的系统保持一致,比如在 CUDA 9.2 上安装:

如果你的系统没有安装 cuda,那么可以通过下面的命令安装 cpu 版本的 PyTorch:

2. 安装 fastai

如果安装过程有什么问题,请确保你的 conda 版本已经更新到最新:

通过 pip 安装

通过下列三步可以用 pip 安装 fastai,要严格安装顺序执行。

1. 安装每日编译 nightly 的 PyTorch,注意 cuda 的版本要和你自己的系统保持一致,比如在 CUDA 9.2 上安装:

2. 安装定制的 torchvision,这个版本是在 nightly-pytorch 上编译的:

3. 安装 fastai:

有些时候,上面的命令会试图安装 torch-0.4.1,这时候可以使用下面的命令:

安装完毕,大家愉快的体验 fastai 的新特性吧!

本站微信群、QQ群(三群号 726282629):

PyTorch入门实战教程

发表回复

您的电子邮箱地址不会被公开。

评论列表(1)

返回顶部