search

microsoft / tensorflow-directml

Fork of TensorFlow accelerated by DirectML
Apache License 2.0
454 stars 32 forks source link

directml on custom tensoflow build ? #381

Closed Akash7789 closed 1 year ago

Akash7789 commented 1 year ago

Is custom tensorflow-directml build possible for tensorflow 2.3.0. If yes, what directml files or libraries would be required.

PatriceVignola commented 1 year ago

Hey @Akash7789 , tensorflow-directml is not possible for tensorflow 2.3.0, but we recently released a preview for tensorflow-directml-plugin, which should work with tensorflow-cpu >= 2.9. If you're interested, you can try it out:

pip install tensorflow-cpu
pip install tensorflow-directml-plugin

PyPi link: https://pypi.org/project/tensorflow-directml-plugin/ GitHub link: https://github.com/microsoft/tensorflow-directml-plugin

If you have any problems, don't hesitate to open an issue on the issues page!