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.
Also, Fardin, advices:
If you have GPUs on your machine and do not want to use them set the environment variable
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
Post a Comment