This repository was archived by the owner on Nov 17, 2023. It is now read-only.
This repository was archived by the owner on Nov 17, 2023. It is now read-only.
MXNet with MKLDNN failed to install horovod #15211
Description
I was able to verify and reproduce the error.
using MXNet built from source is not able to install Horovod
Steps to reproduce:
Using AWS Base DLAMI on P3.2xLarge.
checkout tag 1.5.0.rc0
sudo make -j$(nproc) USE_CUDA=1 USE_CUDA_PATH=/usr/local/cuda USE_CUDNN=1 USE_NCCL=1 USE_BLAS=openblas
cd python
sudo pip install -e .
Able to build successfully and run a simple mnist example.
HOROVOD_GPU_ALLREDUCE=NCCL pip install horovod --user --no-cache-dir -U
error
home/ubuntu/src/incubator-mxnet/python/mxnet/../../include/mkldnn/mkldnn.h:55:28:
fatal error: mkldnn_version.h: No such file or directory
compilation terminated.
INFO: Unable to build MXNet plugin, will skip it.