microsoft / tensorflow-directml-plugin

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

Should I still install tensorflow-cpu instead of tensorflow package after 2.11 or above? #319

Closed KichangKim closed 1 year ago

KichangKim commented 1 year ago

Tensorflow 2.11 release note says that both tensorflow and tensorflow-cpu will install same tensorflow-intel package. So can I use directml plugin with tensorflow package? or still need to install tensorflow-cpu package?

Thanks.

PatriceVignola commented 1 year ago

Yes, after tensorflow 2.11 you will be able to install the tensorflow package on Windows to use tensorflow-directml-plugin, but it will be the same as using tensorflow-cpu. Note that WSL will still need the tensorflow-cpu package, therefore tensorflow-cpu will stay as a required dependency for the tensorflow-directml-plugin for simplicity purposes.

KichangKim commented 1 year ago

Okay, Thanks.