TrainingArgument does not work on colab

I encountered the same problem when I run the code below.

from transformers import TrainingArguments
training_args = TrainingArguments(“test-trainer”)

ERROR:Using the Trainer with PyTorch requires accelerate>=0.20.1: Please run pip install transformers[torch] or pip install accelerate -U

But installing the 2 mentioned libraries did not resolve.
I don’t know what to do now. Help!

1 Like