Skip to content

Commit

Permalink
[SWDEV-479339/SWDEV-498804] Added AMDSMI Dockerfile (#35)
Browse files Browse the repository at this point in the history
Updated AMDSMI Dockerfile

Signed-off-by: Justin Williams <[email protected]>
  • Loading branch information
amd-juwillia authored Jan 7, 2025
1 parent dc400d9 commit 7323881
Showing 1 changed file with 39 additions and 31 deletions.
70 changes: 39 additions & 31 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -1,31 +1,39 @@
# 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"]
# Use rocm/dev-ubuntu-22.04 as the base image
FROM rocm/dev-ubuntu-22.04

# Set environment variables for build directories and package patterns
ENV BUILD_FOLDER=/home/amdsmi/build
ENV DEB_BUILD="amd-smi-lib*99999-local_amd64.deb"
ENV DEB_BUILD_TEST="amd-smi-lib-tests*99999-local_amd64.deb"

# Set the working directory to /home
WORKDIR /home

# Install necessary system packages
RUN apt update && apt-get install -y git build-essential rpm pkg-config g++ python3 python3-pip python3-wheel python3-setuptools

# Upgrade pip and install cmake and virtualenv using pip
RUN python3 -m pip install --upgrade pip setuptools && \
python3 -m pip install cmake virtualenv

# Clone the AMD SMI repository from GitHub
RUN git clone -b amd-mainline https://github.com/ROCm/amdsmi.git

# Navigate to the amdsmi directory
WORKDIR /home/amdsmi

# Build and Install AMDSMI
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 --allow-downgrades ${BUILD_FOLDER}/${DEB_BUILD} && \
sudo ln -s /opt/rocm/bin/amd-smi /usr/local/bin

# Verify the installation of Python packages related to AMD SMI
RUN python3 -m pip list | grep -E "amd|pip|setuptools"

# Set the entrypoint to bash for interactive use
ENTRYPOINT ["/bin/bash"]

0 comments on commit 7323881

Please sign in to comment.