HuggingFace入门教程--环境搭建

news2025/5/14 16:05:30

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

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

1.使用pycharm新建一个项目

在这里插入图片描述

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

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

2.1 安装python3-venv

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

sudo apt install -y python3-venv

在这里插入图片描述
使用命令python -m venv命令新建一个虚拟环境。

python -m venv hug

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

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

在这里插入图片描述

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

2.3 修改interpreter
在这里插入图片描述

通过上述操作我们就把新建项目的虚拟环境配置好了。

3.安装huggingface所需pip 包

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

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

vim ~/.config/pip/pip.conf

添加以下内容。

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple

extra-index-url=
        http://mirrors.aliyun.com/pypi/simple/
        http://pypi.douban.com/simple
        http://pypi.mirrors.ustc.edu.cn/simple/

[install]
trusted-host=
        pypi.tuna.tsinghua.edu.cn
        mirrors.aliyun.com
        pypi.douban.com
        pypi.mirrors.ustc.edu.cn
        download.openmmlab.com

在这里插入图片描述
安装huggingface所需pip包。

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

在这里插入图片描述

4.运行一个模型

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

from transformers import pipeline

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

print(output)

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