# Can I use CUDA with Trainer.train?

**URL:** </static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fcan-i-use-cuda-with-trainer-train%2F17566%3E%5C
**Category:** Beginners\
**Created:** [May 5, 2022, 3:19pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fcan-i-use-cuda-with-trainer-train%2F17566 "2022-05-05T15:19:11Z")\
**Posts on this page:** 1\
**Showing post:** 2

<div class="post-metadata">

**Author:** ![mariosasko](https://sea2.discourse-cdn.com/hellohellohello/user_avatar/discuss.huggingface.co/mariosasko/32/31548_2.png) [@mariosasko](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fmariosasko)%5C
**Post date:** [May 6, 2022, 12:04pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fcan-i-use-cuda-with-trainer-train%2F17566%2F2 "2022-05-06T12:04:52Z")

</div>

Hi! The first error is a known bug (see [Sending a Dataset or DatasetDict to a GPU](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fsending-a-dataset-or-datasetdict-to-a-gpu%2F17208)). Can you update your installation of transformers with `pip install -U transformers` and rerun the first snippet?

---

_[View the full topic](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fcan-i-use-cuda-with-trainer-train%2F17566)._