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:
parent
1344726936
commit
30a3cd75cf
6 changed files with 35 additions and 41 deletions
|
|
@ -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
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue