在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/
版权声明:本博客文章均系本人原创,转载请注名出处