How can I only use CPU insteal of GPU on MBP M1 Pro?

Question

dkjdjdfdskln OP

Created Nov ’21

Replies 1

Boosts 0

Views 1.3k

Participants 2

as above.

I found that M1 Chip has extremely bad performance on training RNNs (158s vs 6h). Could anyone know why the M1 Pro chip is so unfriendly for RNNs? How can I only use CPU to run my code as mlcompute package cannot be recognized somehow.

Boost

Answer 1

cyrusomega OP

Nov ’21

To disable the GPU completely on the M1 use tf.config.experimental.set_visible_devices([], 'GPU'). To disable the GPU for certain operations, use:

with tf.device('/cpu:0'):
    # tf calls here

Someone else will have to answer the questions on RNNs and mlcompute.

0