vast.ai中的Keras多GPU

Question

我试图使用多个GPU在vast.ai上运行一个keras模型。为此，我使用的是keras.utils.multi_gpu_model，但是一直存在以下错误：

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

AttributeError:模块'tensorflow_core._api.v2.config‘没有属性'experimental_list_devices')

我使用的是这个默认的码头：

用于深度学习框架TensorFlow的正式对接图像

成功加载tensorflow/tensorflow:夜-GPU-py3

我还检查了可用的GPU，所有GPU都被正确检测到：

有什么想法吗？

干杯

Answer 1

解决方案：

最后我自己找到了解决办法。我刚刚使用了另一个带有较旧版本的tensorflow (2.0.0)的对接器映像，错误消失了。