Closed xxxpsyduck closed 3 years ago
Welcome to Apache MXNet (incubating)! We are on a mission to democratize AI, and we are glad that you are contributing to it by opening this issue. Please make sure to include all the relevant context, and one of the @apache/mxnet-committers will be here shortly. If you are interested in contributing to our project, let us know! Also, be sure to check out our guide on contributing to MXNet and our development guides wiki.
cc @Kh4L. Serge, would you be able to help on this? Let @leezu know in case you need any information regarding cmake.
Hi @xxxpsyduck
Have you built onnx and onnx_tensorrt as it is done here https://github.com/apache/incubator-mxnet/blob/master/ci/docker/runtime_functions.sh#L553 ?
You need this preprocessor variable to be set https://github.com/apache/incubator-mxnet/blob/master/ci/docker/runtime_functions.sh#L575
@Kh4L hello. Can you explain CI_CUDA_COMPUTE_CAPABILITIES
and CI_CMAKE_CUDA_ARCH
?
You can ignore those and it will compile for all the arch's.
If not, you can check here https://arnon.dk/matching-sm-architectures-arch-and-gencode-for-various-nvidia-cards/ which cuda archs and gencodes to your target devices.
Description
Bug occurs when trying to build mxnet with tensorrt support (Ubuntu 18.04)
Error Message
To Reproduce
Steps to reproduce
(Paste the commands you ran that produced the error.)
What have you tried to solve it?
Environment