llm/SIA
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update docker image

Browse Source
This commit is contained in:
Geens
2025-05-01 17:28:33 +02:00
parent ff6a6ab135
commit d022d41a5d
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
Dockerfile
View File

@@ -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