MXNET CMake 报告 "GPU support is disabled"

Question

我正在尝试在支持 GPU 的机器上从源头赚钱。我有一个 NVIDIA P 100 GPU。当我使用 make 构建 MXNET 时，它报告说：

CMake 日志中的片段如下

-- The CUDA compiler identification is NVIDIA 10.2.89
-- Check for working CUDA compiler: /usr/local/cuda-10.2/bin/nvcc
-- Check for working CUDA compiler: /usr/local/cuda-10.2/bin/nvcc -- works
-- Detecting CUDA compiler ABI info
-- Detecting CUDA compiler ABI info - done

-- GPU support is disabled

我想在 GPU 支持下构建它。我能做什么？

Answer 1

从源构建 MXNet 时，CMake 输出中的消息 GPU support is disabled 有点令人困惑。此特定消息与 CUDA 无关。如果我们尝试启用英特尔 MKL DNN 支持 (-DUSE_MKLDNN=1) 并且系统中未安装英特尔 OpenCL 运行时，它实际上会生成（请参阅其中打印的 CMake file）。 CMake 尝试寻找英特尔 OpenCL 运行时，如果找不到则抛出此消息。我相信 OpenCL 运行时支持 GPU 支持英特尔的 GPU（Intel HD Graphics、Intel Iris 和 Intel Iris Pro）。

如果您不打算使用 Intel 显卡支持，请不要担心此错误消息。它不会影响 CUDA 对 MXNet 的支持。如果您想禁止显示此消息，请禁用 MKL DNN 支持 (-DUSE_MKLDNN=0)。如果您也关心英特尔显卡，请安装 Intel OpenCL Runtime.

MXNET CMake 报告 "GPU support is disabled"

MXNET CMake reports "GPU support is disabled"

gpu

cmake

mxnet