1
$\begingroup$

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

$\endgroup$
2
  • $\begingroup$ Can you include full code that produces this error, including imports? $\endgroup$
    – Dr. Snoopy
    Feb 5, 2020 at 10:56
  • $\begingroup$ 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. $\endgroup$ Feb 5, 2020 at 19:04

1 Answer 1

0
$\begingroup$

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.

$\endgroup$

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service and acknowledge that you have read and understand our privacy policy and code of conduct.

Not the answer you're looking for? Browse other questions tagged or ask your own question.