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 for more information.
1 Like