mac 上keras不能用GPU？

Posted 2023-04-26

我想使用Keras，使用Tensorflow后端的。
我看Tensorflow的说明里只能在linux里开启GPU运行。所以以Tensorflow为后端的Keras在mac下不能用GPU？ 要开的话只能改成Theano后端？

参考技术A 优点：支持python，模型库全，搭模型快，关注度极高，迭代快，可用GPU加速。 缺点： 内部许多类的抽象不合理。 命名略显混乱。 查看中间层输出不够直接。 模型需要compile 这些优缺点很大程度上都是因为现行版本将theano深度耦合，其作者和一些...

Keras/Tensorflow选择GPU/CPU运行

首先，导入os，再按照PCI_BUS_ID顺序，从0开始排列GPU，

import os

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" 

然后就可以选择用哪一个或者那几个GPU运行：

os.environ[”zCUDA_VISIBLE_DEVICES”] = "0"

用0号GPU，即‘/gpu:0‘运行；

os.environ[“CUDA_VISIBLE_DEVICES”] = “0,1”

用0号和1号设备，即‘/gpu:0‘、‘/gpu:1‘运行；

os.environ[“CUDA_VISIBLE_DEVICES”] = “7”

用7号设备，即‘/gpu:7‘运行，但是‘/gpu:7‘根本不存在，于是就只能用CPU运行了。