在Apple Silicon上关闭M1的GPU,仅用CPU进行Tensorflow训练
条评论在苹果M1
系列芯片上运行tensorflow
是可以通过插件tensorflow-metal
进行GPU
训练加速的,并且随着操作系统的升级以及插件的不断完善,M1
的训练性能正在稳步提高,这也是苹果官方推荐的做法。
不过某些情况下,我们还是需要关闭GPU
加速,仅使用CPU
进行训练。那么以下这段代码可以帮助你临时禁用GPU
,而把训练的压力转移到CPU
上来。
import tensorflow as tf |
本文标题:在Apple Silicon上关闭M1的GPU,仅用CPU进行Tensorflow训练
文章作者:牧云踏歌
发布时间:2022-04-18
最后更新:2022-04-18
原始链接:http://www.kankanzhijian.com/2022/04/18/disable-metal-gpu-on-apple-silicon/
版权声明:本博客文章均系本人原创,转载请注名出处