赞
踩
HuggingFace中文直译为”拥抱脸“,是最近非常火爆的一个人工智能社区,官网地址是:https://huggingface.co/ .关于HuggingFace的相关介绍大家可以自行百度。本文主要为刚入人工智能坑的小白指下路,同时也是逼着自己记录下学习过程中的点滴,和热爱学习的小伙伴们分享。
我的环境
为了防止系统中不同项目之间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
通过上述操作我们就把新建项目的虚拟环境配置好了。
由于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
在我们的项目中新建一个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的更深入的知识了。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。