-
Notifications
You must be signed in to change notification settings - Fork 32
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[SWDEV-479339/SWDEV-498804] Added AMDSMI Dockerfile (#35)
Updated AMDSMI Dockerfile Signed-off-by: Justin Williams <[email protected]>
- Loading branch information
1 parent
dc400d9
commit 7323881
Showing
1 changed file
with
39 additions
and
31 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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"] |