# What do I have to do to ensure that a model can run on Gaudi?

**URL:** https://forum.habana.ai/t/what-do-i-have-to-do-to-ensure-that-a-model-can-run-on-gaudi/102
**Category:** FAQ
**Tags:** models
**Created:** [June 30, 2021, 4:28am UTC](https://forum.habana.ai/t/what-do-i-have-to-do-to-ensure-that-a-model-can-run-on-gaudi/102 "2021-06-30T04:28:11Z")
**Posts on this page:** 1
**Page:** 1

<div class="post-metadata">

### Author: ![Greg\_S](https://sea2.discourse-cdn.com/flex020/user_avatar/forum.habana.ai/greg_s/32/33_2.png) [@Greg\_S](https://forum.habana.ai/u/Greg_S)
#### Post date: [June 30, 2021, 4:28am UTC](https://forum.habana.ai/t/what-do-i-have-to-do-to-ensure-that-a-model-can-run-on-gaudi/102/1 "2021-06-30T04:28:12Z")

</div>

Habana’s software integrates TensorFlow or PyTorch, allowing a compatible Python training script running on CPU or GPU to train a model on Gaudi, with the addition of just a few lines of code. See the [Migration Guide](https://docs.habana.ai/en/latest/Migration_Guide/Migration_Guide.html) for more information.