From d022d41a5d54675f18858e0c75aaac2f0cf88c2e Mon Sep 17 00:00:00 2001 From: Geens Date: Thu, 1 May 2025 17:28:33 +0200 Subject: [PATCH] Update docker image --- Dockerfile | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/Dockerfile b/Dockerfile index 4c95bb8..b5ed325 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:12.2.0-runtime-ubuntu22.04 AS base +FROM nvidia/cuda:12.8.1-cudnn-runtime-ubuntu24.04 AS base RUN apt-get update && \ apt-get upgrade -y && \ apt install -y \ @@ -37,7 +37,7 @@ RUN mkdir -p \ # ITB tool setup FROM base AS itb-env -RUN python3 -m venv /root/venvs/itb +RUN python3 -m venv /root/venvs/itb --system-site-packages COPY ./tools/itb/setup.py /root/sia/tools/itb/setup.py RUN /root/venvs/itb/bin/python /root/sia/tools/itb/setup.py egg_info RUN --mount=type=cache,target=/root/.cache/pip \ @@ -46,7 +46,7 @@ RUN rm -rf *.egg-info/ # Train tool setup FROM base AS train-env -RUN python3 -m venv /root/venvs/train +RUN python3 -m venv /root/venvs/train --system-site-packages COPY ./tools/train/setup.py /root/sia/tools/train/setup.py RUN /root/venvs/train/bin/python /root/sia/tools/train/setup.py egg_info RUN --mount=type=cache,target=/root/.cache/pip \ @@ -55,7 +55,7 @@ RUN rm -rf *.egg-info/ # SIA core setup FROM base AS sia-env -RUN python3 -m venv /root/venvs/sia +RUN python3 -m venv /root/venvs/sia --system-site-packages COPY ./setup.py /root/sia/setup.py RUN /root/venvs/sia/bin/python /root/sia/setup.py egg_info RUN sed -i '/\/root\/sia\/lib/d' *.egg-info/requires.txt