跳过导入mxnet上的CUDA库加载

时间:2019-03-18 15:33:50

标签: python mxnet

当仅打算在CPU上进行推理时,是否可以使“ import mxnet”跳过加载CUDA库?

上下文:

  • MXNet已安装CUDA支持,并且模型训练通常在GPU上运行
  • 但是,推断可以在没有GPU的许多节点上并行运行,最好只在CPU上使用相同的MXNet构建,但是某些库在非GPU节点上不可用。

0 个答案:

没有答案