InvalidArgumentError (see above for traceback): Cannot assign a device for operation prefix/Rank: Could not satisfy explicit device specification '/device:GPU:0' because no supported kernel for GPU devices is available.
Closed this issue · 2 comments
When I tried to perform prediction on a single image, I am getting an error at Node prefix/Rank: saying could not satisfy explicit device specification '/device:GPU:0' because no supported kernel for GPU devices is available.
I am trying to solve this problem from quite a long time. Any help or suggestions are appreciated.
I believe you can fix this by passing the argument --gpu-id -1
.
e.g. aocr train ./datasets/training.tfrecords --gpu-id -1
this will let the code to use cpu:0
instead.
see this model.py#L60 and this __main__.py#L88
I believe you can fix this by passing the argument
--gpu-id -1
.
e.g.aocr train ./datasets/training.tfrecords --gpu-id -1
this will let the code to usecpu:0
instead.
see this model.py#L60 and this main.py#L88
i got the same question,need help~.~