Giters

apple / tensorflow_macos

TensorFlow for macOS 11.0+ accelerated using Apple's ML Compute framework.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Tensorflow with gpu

Anusha1408 opened this issue · comments

commented

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

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

commented

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'
commented

@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.