1

I am trying to run a keras model on vast.ai using multiple GPUs. For that I am using keras.utils.multi_gpu_model , however I keep having this error:

if multi_GPU and n_GPUs > 1:
    model = multi_gpu_model(model)

AttributeError: module 'tensorflow_core._api.v2.config' has no attribute 'experimental_list_devices')


I am using this default docker :

Official docker images for deep learning framework TensorFlow

Successfully loaded tensorflow/tensorflow:nightly-gpu-py3

I have also checked the available GPUs and all the GPUs are detected correctly:

Any ideas?

cheers

ebrahimi
  • 1,277
  • 7
  • 20
  • 39
Joaquin Rives
  • 11
  • 1

1 Answers1

0

Solution:

Finally I found the solution myself. I just used another docker image with an older version of tensorflow (2.0.0), and the error disappeared.

Joaquin Rives
  • 11
  • 1