InvalidArgumentError- был явно назначен для / device: GPU: 1, но доступны следующие устройства: [/ job: localhost / replica: 0 / task: 0 / device: CPU: 0, - PullRequest
Новая
       102

InvalidArgumentError- был явно назначен для / device: GPU: 1, но доступны следующие устройства: [/ job: localhost / replica: 0 / task: 0 / device: CPU: 0,

0 голосов
/ 03 апреля 2020

Любая помощь будет оценена. Я новичок в тензорном потоке и программировании в целом. Я следую инструкции в github (https://github.com/experiencor/keras-yolo3), чтобы научиться обнаруживать объекты с помощью YOLO-3. после запуска кода ниже:

!python train.py -c config.json

И я получил следующую ошибку:

tensorflow.python.framework.errors_impl.InvalidArgumentError: Cannot assign a device for` operation replica_1/lambda_6/Shape: node replica_1/lambda_6/Shape (defined at /home/arasdar/anaconda3/lib/python3.7/site-packages/tensorflow_core/python/framework/ops.py:1748) was explicitly assigned to /device:GPU:1 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0, /job:localhost/replica:0/task:0/device:GPU:0, /job:localhost/replica:0/task:0/device:XLA_CPU:0, /job:localhost/replica:0/task:0/device:XLA_GPU:0 ]. Make sure the device specification refers to a valid device. [[replica_1/lambda_6/Shape]]

Я предполагал изменить все "GPU: 1 и от GPU: 2 "до GPU: 0 в файле" ops.py "(указан выше) может исправить эту ошибку, но это не так!

...