在苹果M1系列芯片上运行tensorflow是可以通过插件tensorflow-metal进行GPU训练加速的,并且随着操作系统的升级以及插件的不断完善,M1的训练性能正在稳步提高,这也是苹果官方推荐的做法。

不过某些情况下,我们还是需要关闭GPU加速,仅使用CPU进行训练。那么以下这段代码可以帮助你临时禁用GPU,而把训练的压力转移到CPU上来。

import tensorflow as tf

tf.config.set_visible_devices([], 'GPU')