首页 > 解决方案 > 我应该如何安装和导入 simpletransformers?

问题描述

我一直在使用以下 conda & python 版本:

康达版本:4.6.14

康达构建版本:3.17.8

蟒蛇版本:3.7.3.final.0

我以下列方式安装了 simpletransformers:

conda create -n simpletransformers python pandas tqdm
conda activate simpletransformers

conda install pytorch cpuonly -c pytorch
conda install -c anaconda scipy
conda install -c anaconda scikit-learn
pip install transformers
pip install seqeval
pip install tensorboardx

pip install simpletransformers

这样做之后,我一直在尝试导入分类模型,但运气不佳:

import simpletransformers

我收到以下错误:

ModuleNotFoundError:没有名为“simpletransformers”的模块

有人可以指出我哪里出错了吗?我使用 PyCharm 作为我的 IDE。

标签: pythonpipanacondaconda

解决方案


安装文档适用于我在Mac 和 Ubuntu 上使用 Anaconda:

  1. 安装Anaconda或 Miniconda
  2. 创建一个新的虚拟python 3.7环境并安装pandas和tqdm
    conda create -n simplet python=3.7 pandas tqdm
    conda activate simplet
  3. PyTorch
    3 一个。GPU(use_cuda=True在您的模型中):conda install pytorch cudatoolkit=10.1 -c pytorch
    3 b。CPU(use_cuda=False在您的模型中):conda install pytorch cpuonly -c pytorch
  4. 如果您想在 NVIDIA GPU 上使用 fp16 训练,请安装apex(不要使用pip
  5. 安装简单的变压器。
    pip install simpletransformers

推荐阅读