gpt-llm-trainer icon indicating copy to clipboard operation
gpt-llm-trainer copied to clipboard

which GPU?

Open fredzannarbor opened this issue 2 years ago • 1 comments

I have Colab+ and avalable to me are

v100, t4, tpu

Update: TPU doesn't work using the off-the-shelf version of the script as it assumes an NVIDIA GPU.

which should I use/which will be fastest?

fredzannarbor avatar Aug 17 '23 06:08 fredzannarbor

V100 preferrably. Always better suited for ML pipelines in general

pronoym99 avatar Aug 19 '23 07:08 pronoym99