GPU not working while training a new model in Colab

stanford-futuredata / ColBERT

ColBERT: state-of-the-art neural search (SIGIR'20, TACL'21, NeurIPS'21, NAACL'22, CIKM'22, ACL'23, EMNLP'23)

MIT License

2.67k stars 355 forks source link

GPU not working while training a new model in Colab #317

Open RayRama opened 4 months ago

RayRama commented 4 months ago

When I try to run the available Colab code from this (indexing and search) everything runs smoothly, the GPU also runs properly.

But when I try to train or fine-tune a new model in this colab for some reason the GPU is not working properly (only the RAM is working). Then what should I do?

localcitizen commented 3 weeks ago

@RayRama have you managed to run it? I think the problem might be in Colab resourced allocation.