ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 08:29:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bumped docker OS, python version, torch version

Browse Source
This commit is contained in:
Jaret Burkett
2025-12-23 19:44:20 -07:00
parent 57a2ab1299
commit ed36edd85b
2 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
docker/Dockerfile
View File

@@ -1,4 +1,4 @@
FROM nvidia/cuda:12.8.1-devel-ubuntu22.04 FROM nvidia/cuda:12.8.1-devel-ubuntu24.04
LABEL authors="jaret" LABEL authors="jaret"
@@ -15,7 +15,7 @@ RUN apt-get update && apt-get install --no-install-recommends -y \
build-essential \ build-essential \
cmake \ cmake \
wget \ wget \
python3.10 \ python3.12 \
python3-pip \ python3-pip \
python3-dev \ python3-dev \
python3-setuptools \ python3-setuptools \
@@ -49,7 +49,7 @@ WORKDIR /app
RUN ln -s /usr/bin/python3 /usr/bin/python RUN ln -s /usr/bin/python3 /usr/bin/python
# install pytorch before cache bust to avoid redownloading pytorch # install pytorch before cache bust to avoid redownloading pytorch
RUN pip install --pre --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128 RUN pip install --no-cache-dir torch==2.9.1 torchvision==0.24.1 torchaudio==2.9.1 --index-url https://download.pytorch.org/whl/cu128 --break-system-packages
# Fix cache busting by moving CACHEBUST to right before git clone # Fix cache busting by moving CACHEBUST to right before git clone
ARG CACHEBUST=1234 ARG CACHEBUST=1234
@@ -61,10 +61,12 @@ RUN echo "Cache bust: ${CACHEBUST}" && \
WORKDIR /app/ai-toolkit WORKDIR /app/ai-toolkit
# copy requirements.txt to leverage docker cache
COPY requirements.txt /app/ai-toolkit/requirements.txt
# Install Python dependencies # Install Python dependencies
RUN pip install --no-cache-dir -r requirements.txt && \ RUN pip install --no-cache-dir --break-system-packages -r requirements.txt && \
pip install --pre --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128 --force && \ pip install setuptools==69.5.1 --no-cache-dir --break-system-packages
pip install setuptools==69.5.1 --no-cache-dir
# Build UI # Build UI
WORKDIR /app/ai-toolkit/ui WORKDIR /app/ai-toolkit/ui

3
requirements.txt
View File

@@ -35,4 +35,5 @@ python-slugify
opencv-python opencv-python
pytorch-wavelets==1.3.0 pytorch-wavelets==1.3.0
matplotlib==3.10.1 matplotlib==3.10.1
setuptools==69.5.1 setuptools==69.5.1
scipy==1.12.0