search

apple / tensorflow_macos

TensorFlow for macOS 11.0+ accelerated using Apple's ML Compute framework.
Other
3.66k stars 308 forks source link

Simple way to turn off acceleration #165

Closed karlglazebrook closed 3 years ago

karlglazebrook commented 3 years ago

Hi there,

Is there a simple way to turn off the ML_Compute acceleration and revert to 'stock' tensorflow?

This would be for benchmarking, but also checking for issues with resukts

salsa-sound commented 3 years ago

If I understand the documentation from the readme file. It should be something to do with this: TF_DISABLE_MLC_EAGER=“;Op1;Op2;...” I believe that you can use that to turn off acceleration for specific ops. Not sure if there is a blanket way of doing this for all

anna-tikhonova commented 3 years ago

Thank you for reporting this. We will provide a simple way to turn off acceleration in the upcoming update.

karlglazebrook commented 3 years ago

Thanks that would be welcome! It seems official tensorflow has not yet caught up with arm64 (based on that 'pip install' does not work yet).

ManuelSchneid3r commented 3 years ago

If 0.1-alpha3 fixes this issue, how can we disable acceleration then?