【Hugging Face中文教程】1. 1 安装工作环境

1,546 阅读1分钟

安装

安装一个工作环境

我们将使用两种方式来安装你的工作环境:使用Colab或者Python虚拟环境。对于初学者,我们强烈建议你使用Colab。

使用Google Colab Notebook

如果你不熟悉Colab,我们建议你从以下介绍开始。Colab允许你使用一些硬件:例如GPUs和TPUs,最重要的是,它是免费的。

一旦你已经适应了在Colab中进行操作,创建一个新的笔记本并开始设置: image-20210621225056097.png 使用以下指令安装transformers:

!pip install transformers

然后使用以下方式确认transformers库是否被正确安装:

import transformers

以上安装的仅仅是transformers,而像机器学习框架pytorch、tensorflow是没有被安装的。因为我们使用transformers库的许多不同特性,所以我们建议安装开发版本,它包含了几乎可以想象到的用到的所需的依赖。

!pip install transformers[sentencepiece]

这将花费一些时间,但之后你就可以准备好继续学习剩下的课程了!

使用Python虚拟环境

# 在任意路径下创建文件夹
mkdir transformers-course
# 进入到该文件夹
cd transformers-course
# 创建虚拟环境
python -m venv .env
# 查看
ls -a
# 激活虚拟环境
source .env/bin/activate
# 退出虚拟环境
source .env/bin/deactivate
# 查看python的位置
which python
# 安装transformers
pip install "transformers[sentencepiece]"