1. 安装相应的conda虚拟环境,注意需要安装
1.1 cudatoolkit cudnn pytorch
虚拟环境conda安装cuda11.7和cudnn和pytorch_pytorch cuda11.7-CSDN博客
1.2 transformers
transformers/README_zh-hans.md at main · huggingface/transformers · GitHub
如果有缺失包,如chardet,使用pip install chardet安装
2.验证安装成功
>>> from transformers import pipeline# 使用情绪分析流水线 >>> classifier = pipeline('sentiment-analysis') >>> classifier('We are very happy to introduce pipeline to the transformers repository.') [{'label': 'POSITIVE', 'score': 0.9996980428695679}]
3.下载的模型存放在~/.cache/huggingface/hub中,添加环境变量修改存储路径
sudo gedit ~/.bashrc
export HF_HOME="/path/hf_cache"
source ~/.bashrc
然后重新输入2中的代码,查看存储路径中保存了模型。