全新conda环境安装transformers全流程记录

人工智能110

由于需要安装huggingface的transformers库,但是在我windows的一个已经安装好pytorch和tensorflow的环境中安装,总是出现各种问题。因此在实验室服务器中重新安装。

创建虚拟环境

首先,先创建一个新的虚拟环境,我的名字为pydnlp,选了3.8.10的python。

conda create -n pydnlp python=3.8.10

安装pytorch

指导上提示需要安装tensorflow和pytorch其中之一,当然也可以两个都安装。我先安装pytorch(但是我记得之前运行,需要两个都安装)。

为了使安装的pytorch支持gpu,我们可以使用命令:

nvcc -V

这个过程可能需要很长时间,或者可以用国内的镜像安装(应该),我没用。

[En]

This process may take a long time, or you can use a domestic image installation (should), I am useless.

在这里安装后,您可以使用以下语句删除一些无用的安装包。可查看具体参数。

[En]

After installation here, you can use the following statement to delete some useless installation packages. Specific parameters can be checked.

conda clean -参数

安装transformers

直接使用

pip install transformers
from transformers import pipeline
print(pipeline('sentiment-analysis')('we love you'))

但是很可能报错 connection error,这应该能通过保存模型到本地解决。

本来觉得安装会比较麻烦,因为我之前在一个复杂的已经安装好pytorch和tensorflow的环境上安装运行总是会报一些错。所以特意想记录一下,现在看没有记录的必要。

但是最主要的是怎么用transformers,这几天再探究一下再记录,这次就水个博客。

Original: https://blog.csdn.net/qq_43219379/article/details/123129973
Author: eecspan
Title: 全新conda环境安装transformers全流程记录