我在几个月前写了一段代码来训练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是我的完整训练代码。
1条答案
按热度按时间gojuced71#
我认为它只在Pytorch 2.0.0版本上提供。用1.8试试