首页ubuntu安装transfomers库

ubuntu安装transfomers库

时间: 2025-04-29 18:55:43 浏览: 23

### 安装 Hugging Face Transformers 库为了在 Ubuntu 系统上成功安装 Hugging Face Transformers Python 库，需先确认 Python 和 pip 已经正确安装。可以通过执行如下命令来完成这一步骤： ```bash sudo apt-get update sudo apt-get install -y python3 python3-pip ``` 创建并激活虚拟环境是一个推荐的做法，有助于管理项目依赖关系[^1]。接着，通过 pip 来安装 `transformers` 及其依赖项。打开终端输入以下指令： ```bash pip3 install --upgrade pip pip3 install transformers ``` 如果希望更改默认缓存路径以节省空间或者更好地组织文件结构，在 `.bashrc` 文件中设置环境变量是一种有效的方法。具体操作为编辑该配置文件加入自定义路径： ```bash sudo gedit ~/.bashrc export HF_HOME="/path/hf_cache" ``` 上述 `/path/hf_cache` 需要替换为实际想要使用的存储位置[^3]。最后，在主目录下创建一个新的 Python 脚本用于测试安装效果。此脚本将导入必要的模块并尝试加载预训练模型实例化对象来进行简单预测任务验证： ```python from transformers import AutoTokenizer, AutoModelForSeq2SeqLM from transformers import pipeline import torch import streamlit as st tokenizer = AutoTokenizer.from_pretrained('t5-small') model = AutoModelForSeq2SeqLM.from_pretrained('t5-small') text_input = "translate English to German: Hello world!" input_ids = tokenizer(text_input, return_tensors="pt").input_ids outputs = model.generate(input_ids) print(tokenizer.decode(outputs[0], skip_special_tokens=True)) ``` 以上代码片段展示了如何利用 T5 模型实现英德翻译功能作为示例[^4]。

阅读全文