TVMError: Binary was created using cuda but a loader of that name is not registered

Question

I'm generating tvm and istalling and using it in another docker container

TVM wheel generation code:

    # install cmake
    RUN yum install -y wget
    RUN wget https://github.com/Kitware/CMake/releases/download/v3.20.2/cmake-3.20.2.tar.gz -O cmake.tar.gz && \
        mkdir cmake && tar -zxvf cmake.tar.gz -C cmake --strip-components 1 && \
        cd cmake && ./bootstrap -- -DCMAKE_USE_OPENSSL=OFF && make -j8 && make install && cd .. && \
        rm cmake.tar.gz && rm -rf cmake
    
    RUN echo "CUDA $(ls /lib64/ | grep cu)" > /tmp/build_2.output
    
    # install tvm
    RUN git clone --recursive https://github.com/apache/incubator-tvm tvm && \
        cd tvm && \
        git reset --hard 338940dc5044885412f9a6045cb8dcdf9fb639a4 && \
        git submodule init && \
        git submodule update && \
        mkdir ./build && \
        cp cmake/config.cmake build && \
        cd build && \
        cmake --debug-output -DUSE_CUDA=ON -DUSE_CUDNN=ON -DUSE_CUBLAS=ON -DUSE_THRUST=ON -DUSE_LLVM=ON .. &>> /tmp/build_2.output && \
        make -j$(nproc) &>> /tmp/build_2.output && \
        cd ../python && python3.8 setup.py bdist_wheel

Verified that cuda .so files were created in /tmp/build_2.output

While installing and running the tvm in another place getting below error:

tvm.runtime.load_module

*TVMError: Binary was created using cuda but a loader of that name is not registered. Available loaders are GraphRuntimeFactory, metadata, GraphExecutorFactory, VMExecutable. Perhaps you need to recompile with this runtime enabled.*

How to register cuda loader?