Imageai: AMD GPU support via PlaidML?

Created on 8 Mar 2019  路  3Comments  路  Source: OlafenwaMoses/ImageAI

would it be possible to use this library with PlaidML to support more GPUs? https://github.com/plaidml/plaidml

I tried to add

os.environ["KERAS_BACKEND"] = "plaidml.keras.backend"

but getting

imageai\Detection\keras_retinanet\backend\dynamic.py", line 12, in <module>
    assert _backend in backends
AssertionError

Most helpful comment

This will be considered and may be added in future versions.

All 3 comments

This will be considered and may be added in future versions.

Also interested ! Is there any workaround we can set up in the mean time @OlafenwaMoses ?
Thanks !

I'm interested too... Without an Nvidia GPU I'd have to pay for GCP or something similar but it would be nice to take advantage of my existing AMD one. Thanks!

Was this page helpful?
0 / 5 - 0 ratings