32 строки
1.0 KiB
Docker
32 строки
1.0 KiB
Docker
|
# 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"]
|