python ImportError:使用`Trainer`和`PyTorch`需要`accelerate`

ruarlubt  于 2023-05-21  发布在  Python
关注(0)|答案(1)|浏览(945)

我在几个月前写了一段代码来训练NER模型,它运行得很好。然而,同样的代码现在给了我这个错误:

ImportError: Using the `Trainer` with `PyTorch` requires `accelerate`: Run `pip install --upgrade accelerate`

我尝试按照指令here安装accelerate,但它没有运行,这是我尝试运行accelerate时的输出:

2023-05-15 07:28:08.310904: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT

Copy-and-paste the text below in your GitHub issue

- `Accelerate` version: 0.19.0
- Platform: Linux-5.15.107+-x86_64-with-glibc2.31
- Python version: 3.10.11
- Numpy version: 1.22.4
- PyTorch version (GPU?): 2.0.0+cu118 (True)
- System RAM: 12.68 GB
- GPU type: Tesla T4
- `Accelerate` default config:
    - compute_environment: LOCAL_MACHINE
    - distributed_type: NO
    - mixed_precision: fp16
    - use_cpu: False
    - num_processes: 1
    - machine_rank: 0
    - num_machines: 1
    - rdzv_backend: static
    - same_network: False
    - main_training_function: main
    - downcast_bf16: False
    - tpu_use_cluster: False
    - tpu_use_sudo: False

我也安装了TensorRT,但这并没有解决这个问题。
Here是我的完整训练代码。

gojuced7

gojuced71#

我认为它只在Pytorch 2.0.0版本上提供。用1.8试试

相关问题