python - 跳过在 import mxnet 上加载 CUDA 库

Question

当只打算在 CPU 上进行推断时，是否可以让“import mxnet”跳过加载 CUDA 库？

语境：

• MXNet 安装了 CUDA 支持，模型训练通常在 GPU 上运行
• 然而，推理可以在没有 GPU 的许多节点上并行运行，并且最好使用仅具有 CPU 的相同 MXNet 构建，但某些库在非 GPU 节点上不可用。

标签： pythonmxnet