RuntimeError when running on Colab GPU

Hello,

I am trying to train a model for token classification (NER), more or less following the example on the Huggingface course on Token Classification.

When I attempt to initiate arguments, using transformers.TrainingArguments, as below:

from transformers import TrainingArguments

args = TrainingArguments(
    "bert-finetuned-ner",
    evaluation_strategy="epoch",
    save_strategy="epoch",
    learning_rate=2e-5,
    num_train_epochs=3,
    weight_decay=0.01,
    push_to_hub=True,
)

I get the error:

RuntimeError: Failed to import transformers.training_args because of the following error (look up to see its traceback):
/usr/local/lib/python3.7/dist-packages/_XLAC.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZN2at13_foreach_erf_EN3c108ArrayRefINS_6TensorEEE

Why is this? Is this a bug for transformers on Colab GPUs?

Can you provide a Colab to reproduce this error? What’s your Transformers version?