leezu commented on a change in pull request #18622: URL: https://github.com/apache/incubator-mxnet/pull/18622#discussion_r457594285
########## File path: ci/docker/Dockerfile.build.jetson ########## @@ -90,6 +90,9 @@ RUN wget https://repo.download.nvidia.com/jetson/x86_64/pool/r32.4/c/cuda/cuda-c # while previously it used to store it at /usr/local/cuda/targets/aarch64-linux/lib/stubs RUN ln -s /usr/lib/aarch64-linux-gnu/libcublas.so /usr/local/cuda/targets/aarch64-linux/lib/stubs/libcublas.so +# Need the stub for libcuda.so in a place where host linker can find it to link the unit tests +RUN ln -s /usr/local/cuda/targets/aarch64-linux/lib/stubs/libcuda.so /usr/lib/aarch64-linux-gnu/libcuda.so Review comment: `mxnet_unit_tests` is a cmake target that depends on the `libmxnet` target. The libraries detected when compiling `libmxnet` will also be propagated to `mxnet_unit_tests`. I think the problem is that you just added the `cuda` as dependency, instead of the correct `CUDA::cuda_driver` found by https://gitlab.kitware.com/cmake/cmake/-/blob/master/Modules/FindCUDAToolkit.cmake. ---------------------------------------------------------------- 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: us...@infra.apache.org