How to set gpu device for hugging trainer?
I am trying to set gpu device for HF trainer. By default it uses device:0. I want to use a custom device. I tried usingtorch.cuda.set_device(). It does not work.
2024-09-16T16:18:18Z
Did you figure this out ?