benz725 commented on issue #19731:
URL:
https://github.com/apache/incubator-mxnet/issues/19731#issuecomment-757600292
@leezu @Kh4L
the incubator-mxnet/ci/docker/runtime_functions.sh file,
I can run to the line before the
" cmake -DUSE_CUDA=1 \
-DUSE_CUDNN=1 \
-DUSE_OPENCV=1 \
-DUSE_TENSORRT=1 \
-DUSE_OPENMP=0 \
-DUSE_MKLDNN=0 \
-DUSE_NVML=OFF \
-DUSE_MKL_IF_AVAILABLE=OFF \
-DMXNET_CUDA_ARCH="$CI_CMAKE_CUDA_ARCH" \
-G Ninja \
/work/mxnet " line
But when every time I do cmake , terminal will exit automatically, so I
follow the instructions of the
/mxnet-src/incubator-mxnet/config/linux_gpu.cmake but added THE USE_TENSORRT
flag: $ mkdir build; cd build # $ cmake .. -DUSE_TENSORRT=1# $ cmake --build
. --parallel 32
THE cmake can do correctly, but when do cmake --build . --parallel 32, It
reported:
> In file included from
/home/yfzx/mxnet-src/incubator-mxnet/src/operator/subgraph/tensorrt/./onnx_to_tensorrt.h:31:0,
from
/home/yfzx/mxnet-src/incubator-mxnet/src/operator/subgraph/tensorrt/onnx_to_tensorrt.cc:29:
/home/yfzx/mxnet-src/incubator-mxnet/3rdparty/onnx-tensorrt/NvOnnxParser.h:26:10:
fatal error: NvInfer.h: 没有那个文件或目录
#include "NvInfer.h"
^~~~~~~~~~~
compilation terminated.
CMakeFiles/mxnet.dir/build.make:4813: recipe for target
'CMakeFiles/mxnet.dir/src/operator/subgraph/tensorrt/onnx_to_tensorrt.cc.o'
failed
make[2]: ***
[CMakeFiles/mxnet.dir/src/operator/subgraph/tensorrt/onnx_to_tensorrt.cc.o]
Error 1
CMakeFiles/Makefile2:546: recipe for target 'CMakeFiles/mxnet.dir/all' failed
make[1]: *** [CMakeFiles/mxnet.dir/all] Error 2
Makefile:159: recipe for target 'all' failed
make: *** [all] Error 2
I'm still not sure about what the path /work/mxnet means, does the /work
means the root of the source code or the /work/mxnet means the root of the
source code?
And how to fix my bug?
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
