github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Double tap torch install to force blackwell compatability

Browse Source
This commit is contained in:
Jaret Burkett
2025-06-02 20:22:22 -06:00
parent 85f4b47e79
commit 82b90b902e
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docker/Dockerfile
View File

@@ -63,6 +63,7 @@ WORKDIR /app/ai-toolkit
# Install Python dependencies # Install Python dependencies
RUN pip install --no-cache-dir -r requirements.txt && \ RUN pip install --no-cache-dir -r requirements.txt && \
pip install --pre --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu128 --force && \
pip install setuptools==69.5.1 --no-cache-dir pip install setuptools==69.5.1 --no-cache-dir
# Build UI # Build UI