当前位置:   article > 正文

Transformers之环境安装_transformers安装

transformers安装

Transformers 提供了数以千计的预训练模型,支持 100 多种语言的文本分类、信息抽取、问答、摘要、翻译、文本生成。

Transformers 支持三个最热门的深度学习库: Jax, PyTorch 以及 TensorFlow — 并与之无缝整合。你可以直接使用一个框架训练你的模型然后用另一个加载和推理。

本文主要介绍安装 Transformers、设置缓存,并选择性配置 Transformers 以离线运行。

使用 pip 安装

你应该使用虚拟环境安装 Transformers。使用虚拟环境,你可以轻松管理不同项目,避免不同依赖项之间的兼容性问题。

首先,在项目目录中创建虚拟环境:

conda create -n python-transformers python=3.10
  • 1

激活虚拟环境:

conda activate python-transformers
  • 1

现在你可以使用以下命令安装 Transformers:

pip install transformers
  • 1

若仅需 CPU 支持,可以使用单行命令方便地安装

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/笔触狂放9/article/detail/624406
推荐阅读
相关标签