How can I disable this plugin?

microsoft / tensorflow-directml-plugin

DirectML PluggableDevice plugin for TensorFlow 2

Apache License 2.0

179 stars 23 forks source link

Closed Neizvestnyj closed 1 year ago

Neizvestnyj commented 1 year ago

Since this plugin is unstable, I would like to be able to switch from GPU to CPU. Is it really possible to do this and how?

Neizvestnyj commented 1 year ago

os.environ['DML_VISIBLE_DEVICES'] = "-1"