在没有支持CUDA的情况下，无法在具有GPU的计算机上安装Keras MXNet

作者: 无思
发布时间: 2025-03-09 07:11:51 (8天前)
转自：

        我明确地尝试安装一个没有CUDA支持的mxnet版本。使用cuda支持安装时，我可以在此处运行此示例。我跟随keras＆amp;这里有mxnet安装指南....

2 条回复

0#
回复此人
哎？小查查 | 2019-08-31 10-32
这是因为
model.compile
使用CPU或GPU取决于GPU是否在机器中可用。看起来它不会检查是否安装了GPU版本的MXNet。你可以强迫
model.compile
通过显式指定上下文来使用CPU。例：
```
    
      model.compile(loss=’categorical_crossentropy’,
              optimizer=RMSprop(),
              metrics=[‘accuracy’],
              context=[“cpu()”])
</code>
  
```
编辑

登录后才能参与评论


