Trainer crashes on GCP

While running the same setup as given the [Text classification tutorial](https://github.com/huggingface/setfit/blob/main/notebooks/text-classification.ipynb), my notebook on Google Cloud Platform crashes without any further notice or error while calling the `trainer.train()`.

I tried to watch `nvidia-smi` during crash time but kernel crashes too quickly to notice anything. The machine definitely should be big enough. The specs are:

> 60 GB memory
1x P100 GPU
nvidia-drivers are installed
`torch.cuda.is_available()` returns `True`

The same code does run on Google Colab with no issues.

What can I try to debug? Thanks!