Add missing dependencies

mila-iqia · Jan 21, 2025 · 1c538b0 · 1c538b0
1 parent b2e4cc2
commit 1c538b0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/docker/Dockerfile-cuda b/docker/Dockerfile-cuda
@@ -41,7 +41,7 @@ COPY . /milabench/milabench/
 
 ENV DEBIAN_FRONTEND=noninteractive
 RUN apt-get update -y &&\
-    apt-get install -y --no-install-recommends git build-essential curl python3 python-is-python3 python3-pip libgl1-mesa-glx &&\
+    apt-get install -y --no-install-recommends git build-essential curl python3 python-is-python3 python3-pip libgl1-mesa-glx libglib2.0-0 gcc g++ python3-dev &&\
     curl -o /etc/apt/trusted.gpg.d/mellanox.asc https://content.mellanox.com/ofed/RPM-GPG-KEY-Mellanox &&\
     curl -o /etc/apt/sources.list.d/mellanox.list https://linux.mellanox.com/public/repo/mlnx_ofed/${MOFED_VERSION}/ubuntu22.04/mellanox_mlnx_ofed.list &&\
     apt-get update -y &&\
@@ -52,7 +52,7 @@ RUN apt-get update -y &&\
 # Install Rust
 RUN curl https://sh.rustup.rs -sSf | sh -s -- -y
 ENV PATH="/root/.cargo/bin:${PATH}"
-ENV CUDA_HOME=/usr/local/cuda-11.8
+ENV CUDA_HOME=/usr/local/cuda-12.1
 
 # Install Milabench
 # -----------------

diff --git a/docker/Dockerfile-rocm b/docker/Dockerfile-rocm
@@ -39,7 +39,7 @@ COPY . /milabench/milabench/
 # build-essential: for rust
 
 RUN apt-get update &&\
-    apt-get install -y git build-essential curl python3.10 python-is-python3 python3-pip &&\
+    apt-get install -y git build-essential curl python3.10 python-is-python3 python3-pip libgl1-mesa-glx libglib2.0-0 gcc g++ python3-dev &&\
     apt-get clean &&\
     rm -rf /var/lib/apt/lists/*