Run Horovod with CPU only

I was interested in running federated model training using Horovod in the Kubernetes cluster. The first question I had was - is it possible to run Horovod on CPU, because our containers don't have GPUs.
I found out on GIT repository of Horovod , that this is possible. I have to install within Horovod non-gpu version of Tensorflow, and everything should fly.
Also no need to specify GPU settings in the code, like

config.gpu_options.allow_growth = True
config.gpu_options.visible_device_list = str(hvd.local_rank())

Also, Fardin, advices:

If you have GPUs on your machine and do not want to use them set the environment variable CUDA_VISIBLE_DEVICES=-1.

Comments

Popular posts from this blog

Install Kubeflow locally

RabbitMQ and OpenShift