Модель RNN вызывает ошибку TensorFlow: невозможно объединить устройства с несовместимыми идентификаторами - PullRequest
Новая
       67

Модель RNN вызывает ошибку TensorFlow: невозможно объединить устройства с несовместимыми идентификаторами

0 голосов
/ 26 марта 2020

В чем проблема?

При попытке использовать приведенный здесь пример текущей модели https://github.com/ray-project/ray/blob/master/rllib/examples/custom_keras_rnn_model.py в качестве пользовательской модели с PPO, я получаю следующую ошибку:

tensorflow.python.framework.errors_impl.InternalError: Constraining by assigned device should not cause an error. Original root's assigned device name: /job:localhost/replica:0/task:0/device:GPU:0 node's assigned device name "/job:localhost/replica:0/task:0/device:GPU:1. Error: Cannot merge devices with incompatible ids: '/job:localhost/replica:0/task:0/device:GPU:0' and '/job:localhost/replica:0/task:0/device:GPU:1'

Версия Ray и другая системная информация (Python версия, версия TensorFlow, ОС):

Ray 0.8.2 Python 3.6.10 tenorflow-gpu 2.1.0 mujoco_py 2.0.2.2

...