Пропустить загрузку библиотек CUDA при импорте mxnet - PullRequest
0 голосов
/ 18 марта 2019

Можно ли сделать так, чтобы "import mxnet" пропускал загрузку библиотек CUDA, когда предполагается только вывод на ЦП?

Контекст:

  • MXNet устанавливается с поддержкой CUDA, а тренинги моделей обычно выполняются на графических процессорах
  • , однако вывод можно выполнять параллельно на многих узлах без графического процессора ибыло бы неплохо использовать ту же сборку MXNet только с процессором, но некоторые библиотеки недоступны на узлах без графического процессора.
...