Dockerfile

    # Use ubuntu 20.04 base image
    FROM compute-artifactory.amd.com:5000/rocm-base-images/ubuntu-20.04-bld:2024112501

    # Set environment variables
    ENV BUILD_FOLDER=/src/build
    ENV DEB_BUILD="amd-smi-lib*99999-local_amd64.deb"
    ENV DEB_BUILD_TEST='amd-smi-lib-tests*99999-local_amd64.deb'

    # Set up the working directory
    WORKDIR /src

    # Copy the source code into the container
    COPY . /src

    # Run the build and install commands
    RUN rm -rf ${BUILD_FOLDER} && \
        mkdir -p ${BUILD_FOLDER} && \
        cd ${BUILD_FOLDER} && \
        cmake .. -DBUILD_TESTS=ON -DENABLE_ESMI_LIB=ON && \
        make -j $(nproc) VERBOSE=1 && \
        make package && \
        sudo apt install -y ${BUILD_FOLDER}/${DEB_BUILD} && \
        sudo ln -s /opt/rocm/bin/amd-smi /usr/local/bin

    # Verify installation
    RUN python3 -m pip list | grep amd && \
        python3 -m pip list | grep pip && \
        python3 -m pip list | grep setuptools

    # Set the entrypoint
    ENTRYPOINT ["/bin/bash"]