# How to set gpu device for hugging trainer?

**URL:** </static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fhow-to-set-gpu-device-for-hugging-trainer%2F8327%3E%5C
**Category:** 🤗Transformers\
**Created:** [July 13, 2021, 8:36pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fhow-to-set-gpu-device-for-hugging-trainer%2F8327 "2021-07-13T20:36:09Z")\
**Posts on this page:** 2\
**Page:** 1

<div class="post-metadata">

**Author:** ![rabiulawal](https://sea2.discourse-cdn.com/hellohellohello/user_avatar/discuss.huggingface.co/rabiulawal/32/2271_2.png) [@rabiulawal](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Frabiulawal)%5C
**Post date:** [July 13, 2021, 8:36pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fhow-to-set-gpu-device-for-hugging-trainer%2F8327%2F1 "2021-07-13T20:36:09Z")

</div>

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 using `torch.cuda.set_device()`. It does not work.

---

<div class="post-metadata">

**Author:** ![mayank64ce](https://avatars.discourse-cdn.com/v4/letter/m/0ea827/32.png) [@mayank64ce](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fmayank64ce)%5C
**Post date:** [September 16, 2024, 4:18pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fhow-to-set-gpu-device-for-hugging-trainer%2F8327%2F2 "2024-09-16T16:18:18Z")

</div>

Did you figure this out ?