search

microsoft / tensorflow-directml-plugin

DirectML PluggableDevice plugin for TensorFlow 2
Apache License 2.0
179 stars 23 forks source link

Plugin install via pip forces tensorflow-cpu to be 2.10 #360

Open joshglens opened 1 year ago

joshglens commented 1 year ago

While the readme says that tensorflow 2.12 is supported, when installing this plugin via pip it automatically downgrades to 2.10. Reinstalling tensorflow-cpu v2.12 results in a conflict and does not work.

KiTant commented 11 months ago

I have that problem too

HAL9000COM commented 9 months ago

The package published on PyPI is v0.4.0, before the changes to use TF 2.12. Maybe wait for a v0.5.0 release, if this project is still active.

PatriceVignola commented 8 months ago

We had to downgrade to 2.10 because TensorFlow 2.12 contained some regressions related to plugins that broke a bunch of existing scenarios, so using 2.12 or more recent releases most likely won't work.

Unfortunately, we had to pause the development of this plugin until further notice so it's not something that we can fix for the foreseeable future. For the time being, all latest DirectML features and performance improvements are going into onnxruntime for inference scenarios. We'll update this issue if/when things change.