Tensorflow with gpu

[Anusha1408]

Can anyone tell how to use tensorflow with gpu on mac m1 ?

Is there a GPU available:
[]
Is the Tensor on GPU #0:
False

[bksaini078]

I don't know if i got the question right but , you can try this

from tensorflow.python.framework.ops import disable_eager_execution
disable_eager_execution()

from tensorflow.python.compiler.mlcompute import mlcompute
mlcompute.set_mlc_device(device_name='gpu')# this can be 'any','cpu','gpu'

[IAINATTWATER]

@bksaini078 That's what I use and it appears to work. Activity monitor shows the GPU well used for the Python38 process.FashionMNIST was 30% faster for the sample I produced.