Swati Rawat
108 regels
3.2 KiB
Docker
108 regels
3.2 KiB
Docker
# Use amdsmi base image
|
|
FROM amdsmi-image
|
|
|
|
# Sync ROCm repositories
|
|
RUN apt-get update && apt-get install -y wget gnupg2 && \
|
|
wget -qO - http://repo.radeon.com/rocm/rocm.gpg.key | apt-key add - && \
|
|
echo 'deb [arch=amd64] http://repo.radeon.com/rocm/apt/debian/ jammy main' | tee /etc/apt/sources.list.d/rocm.list
|
|
|
|
# Install ROCm runtime and dependencies
|
|
RUN apt-get update && \
|
|
apt-get install -y \
|
|
rocm-smi-lib \
|
|
cmake \
|
|
make \
|
|
g++ \
|
|
doxygen \
|
|
texlive-latex-base \
|
|
automake \
|
|
unzip \
|
|
build-essential \
|
|
autoconf \
|
|
libtool \
|
|
pkg-config \
|
|
libgflags-dev \
|
|
libgtest-dev \
|
|
clang \
|
|
libc++-dev \
|
|
curl \
|
|
libcap-dev \
|
|
python3-argcomplete \
|
|
python3-pip \
|
|
bash && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
# Update setuptools
|
|
RUN python3 -m pip install --upgrade setuptools==59.6.0
|
|
|
|
# Check for modprobe
|
|
RUN command -v modprobe || echo "modprobe not found"
|
|
|
|
# Set environment variables
|
|
ENV GRPC_ROOT=/opt/grpc
|
|
ENV RDC_LIB_DIR=/opt/rocm/lib/rdc
|
|
ENV CMAKE_ROOT=/usr/bin/cmake
|
|
|
|
# Install gRPC
|
|
RUN git clone -b v1.67.1 https://github.com/grpc/grpc --depth=1 --shallow-submodules --recurse-submodules && \
|
|
cd grpc && \
|
|
cmake -B build \
|
|
-DgRPC_INSTALL=ON \
|
|
-DgRPC_BUILD_TESTS=OFF \
|
|
-DBUILD_SHARED_LIBS=ON \
|
|
-DCMAKE_INSTALL_PREFIX="$GRPC_ROOT" \
|
|
-DCMAKE_INSTALL_LIBDIR=lib \
|
|
-DCMAKE_BUILD_TYPE=Release && \
|
|
make -C build -j $(nproc) && \
|
|
make -C build install && \
|
|
echo "$GRPC_ROOT" | tee /etc/ld.so.conf.d/grpc.conf
|
|
|
|
# Build and install RDC
|
|
RUN git clone https://github.com/ROCm/rocm-systems --recursive && \
|
|
cd rocm-systems/projects/rdc && \
|
|
cmake -B build -DGRPC_ROOT="$GRPC_ROOT" \
|
|
-DSMIDIR="$SMI_DIR" \
|
|
-DBUILD_TESTS=OFF \
|
|
-DBUILD_PROFILER=OFF \
|
|
-DBUILD_RUNTIME=OFF \
|
|
-DBUILD_RVS=OFF && \
|
|
make -C build -j $(nproc) && \
|
|
make -C build install
|
|
|
|
# Update system library path
|
|
RUN export RDC_LIB_DIR=/opt/rocm/lib/rdc && \
|
|
export GRPC_LIB_DIR="/opt/grpc/lib" && \
|
|
echo "${RDC_LIB_DIR}" | tee /etc/ld.so.conf.d/x86_64-librdc_client.conf && \
|
|
echo "${GRPC_LIB_DIR}" | tee -a /etc/ld.so.conf.d/x86_64-librdc_client.conf && \
|
|
ldconfig
|
|
|
|
# Expose the port for Prometheus
|
|
EXPOSE 50051
|
|
|
|
# Expose the prometheus plugin port
|
|
EXPOSE 5000
|
|
|
|
# Set working directory to /opt/rocm/libexec/rdc/python_binding
|
|
WORKDIR /opt/rocm/libexec/rdc/python_binding
|
|
|
|
# Install Prometheus
|
|
RUN wget https://github.com/prometheus/prometheus/releases/download/v2.41.0/prometheus-2.41.0.linux-amd64.tar.gz && \
|
|
tar xvfz prometheus-2.41.0.linux-amd64.tar.gz && \
|
|
mv prometheus-2.41.0.linux-amd64/prometheus /usr/local/bin/ && \
|
|
rm -rf prometheus-2.41.0.linux-amd64*
|
|
|
|
# Install Grafana
|
|
RUN wget https://dl.grafana.com/oss/release/grafana-9.3.2.linux-amd64.tar.gz && \
|
|
tar -zxvf grafana-9.3.2.linux-amd64.tar.gz && \
|
|
mv grafana-9.3.2 /usr/local/grafana && \
|
|
rm grafana-9.3.2.linux-amd64.tar.gz
|
|
|
|
# Install Prometheus client for Python
|
|
RUN python3 -m pip install prometheus_client
|
|
|
|
# Ensure sudo can run without a password for the rdcd command
|
|
RUN echo "ALL ALL=(ALL) NOPASSWD: /opt/rocm/bin/rdcd" >> /etc/sudoers
|
|
|
|
# Set the entry point to run the rdcd command
|
|
ENTRYPOINT ["sudo", "/opt/rocm/bin/rdcd", "-u"]
|