> 文章列表 > HuggingFace入门教程--环境搭建

HuggingFace入门教程--环境搭建

HuggingFace入门教程--环境搭建

HuggingFace中文直译为”拥抱脸“,是最近非常火爆的一个人工智能社区,官网地址是:https://huggingface.co/ .关于HuggingFace的相关介绍大家可以自行百度。本文主要为刚入人工智能坑的小白指下路,同时也是逼着自己记录下学习过程中的点滴,和热爱学习的小伙伴们分享。
HuggingFace入门教程--环境搭建
我的环境

  • ubuntu2204
  • python3.10 (ubuntu2204默认版本)
  • IDE pycharm 2023.1

1.使用pycharm新建一个项目

HuggingFace入门教程--环境搭建
HuggingFace入门教程--环境搭建
HuggingFace入门教程--环境搭建

2.为新项目配置虚拟环境

为了防止系统中不同项目之间pip包的相互污染,推荐大家在新建python项目时都使用虚拟环境来操作。

2.1 安装python3-venv

创建虚拟环境需要使用python3-venv,我们先安装一下。

sudo apt install -y python3-venv

HuggingFace入门教程--环境搭建
使用命令python -m venv命令新建一个虚拟环境。

python -m venv hug

上述命令新建了一个名为hug的虚拟环境,等右下角的新建进度条走完之后,你会在目录中看到hug文件夹,展开它进入bin目录,会发现在bin目录下有许多对虚拟环境操作的命令。
HuggingFace入门教程--环境搭建
HuggingFace入门教程--环境搭建2.2 激活虚拟环境。

目前我们已经新建了一个名为hug的虚拟环境,进入hug文件夹使用source命令激活hug虚拟环境,也可以使用source + activate所在路径的方式激活,请参照下图。

HuggingFace入门教程--环境搭建

通过上述操作说明我们的虚拟环境已经建好了(截图中的字写错了,应该是激活虚拟环境)。

2.3 修改interpreter
HuggingFace入门教程--环境搭建
HuggingFace入门教程--环境搭建
HuggingFace入门教程--环境搭建
通过上述操作我们就把新建项目的虚拟环境配置好了。

3.安装huggingface所需pip 包

由于pip默认从国外下载pip包,为了加速pip包的下载,我们需要将pip下载源更改为国内的。

修改pip.conf文件,如果没有就新建。

vim ~/.config/pip/pip.conf

添加以下内容。

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simpleextra-index-url=http://mirrors.aliyun.com/pypi/simple/http://pypi.douban.com/simplehttp://pypi.mirrors.ustc.edu.cn/simple/[install]
trusted-host=pypi.tuna.tsinghua.edu.cnmirrors.aliyun.compypi.douban.compypi.mirrors.ustc.edu.cndownload.openmmlab.com

HuggingFace入门教程--环境搭建
安装huggingface所需pip包。

pip install datasets evaluate transformers[sentencepiece]
pip install torch pipeline

HuggingFace入门教程--环境搭建

4.运行一个模型

在我们的项目中新建一个PipeDemo1.py文件,并添加以下内容。

from transformers import pipelineclassifier = pipeline("sentiment-analysis")
output = classifier("I've been waiting for a HuggingFace course my whole life.")print(output)

运行PipeDemo1.py程序,第一次运行时需要先下载相应的模型到本地,以后再运行该程序时就不需要重新下载模型了。
HuggingFace入门教程--环境搭建
如果你能顺利的出现上图说明你的HuggingFace的开发环境已经配置好了,接下来就可以进一步的学习HuggingFace的更深入的知识了。

HuggingFace入门教程--环境搭建

浏阳门户网