跳过在导入mxn上加载CUDA库

2024-04-26 09:30:19 发布

您现在位置:Python中文网/ 问答频道 /正文

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

上下文:

  • MXNet安装有CUDA支持,模型培训通常在GPU上运行
  • 但是,推理可以在没有GPU的情况下在多个节点上并行运行,最好只使用带CPU的同一MXNet构建,但有些库在非GPU节点上不可用

Tags: 模型import节点gpu情况cpumxnetcuda