CODEWITHSUNDEEP
tensorflowamazon-sagemakerdistributed-traininghorovod
juvchan
juvchan

Reputation: 6255

Can Horovod with TensorFlow work on non-GPU instances in Amazon SageMaker?

I want to perform distributed training on Amazon SageMaker. The code is written with TensorFlow and similar to the following code where I think CPU instance should be enough:  https://github.com/horovod/horovod/blob/master/examples/tensorflow_word2vec.py

Can Horovod with TensorFlow work on non-GPU instances in Amazon SageMaker?

Upvotes: 0

Views: 135

Answers (1)

Arun Lokanatha
Arun Lokanatha

Reputation: 510

Yeah you should be able to use both CPU's and GPU's with Horovod on Amazon SageMaker. Please follow the below example for the same

https://github.com/aws/amazon-sagemaker-examples/blob/main/sagemaker-python-sdk/tensorflow_script_mode_horovod/tensorflow_script_mode_horovod.ipynb

Upvotes: 0

Related Questions