Start: Migrate options from cu121/118 to cu12

This encapsulates more cuda versions and makes install easier for
new users.

Signed-off-by: kingbri <8082010+kingbri1@users.noreply.github.com>
This commit is contained in:
kingbri 2025-08-19 22:25:30 -04:00
parent 1344726936
commit 30a3cd75cf
6 changed files with 35 additions and 41 deletions

View file

@ -26,8 +26,8 @@ WORKDIR /app
# Get requirements
COPY pyproject.toml .
# Install packages specified in pyproject.toml cu121, extras
RUN pip install --no-cache-dir .[cu121,extras]
# Install packages specified in pyproject.toml cu12, extras
RUN pip install --no-cache-dir .[cu12,extras]
RUN rm pyproject.toml