Fix cuda images

aidotse · Dec 16, 2024 · 8892284 · 8892284
1 parent 603489a
commit 8892284
Show file tree

Hide file tree

Showing 5 changed files with 492 additions and 16 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -6,7 +6,7 @@ RUN pip install --upgrade pip && \
     poetry config virtualenvs.in-project true
 
 COPY pyproject.toml poetry.lock /app/
-RUN poetry install --no-ansi --with=cuda
+RUN poetry install --no-ansi --with cuda -E cuda
 
 FROM python:3.11-slim-bookworm
 WORKDIR /app

diff --git a/Dockerfile.cpu b/Dockerfile.cpu
@@ -6,7 +6,7 @@ RUN pip install --upgrade pip && \
     poetry config virtualenvs.in-project true
 
 COPY pyproject.toml poetry.lock /app/
-RUN poetry install --no-ansi --with=cpu
+RUN poetry install --no-ansi --sync
 
 FROM python:3.11-slim-bookworm
 WORKDIR /app

diff --git a/README.md b/README.md
@@ -61,7 +61,11 @@ project. Follow their installation instructions if you do not have it.
 
 ```bash
 # Install the dependencies
-poetry install --with=<cuda or cpu>
+poetry install
+# OR
+poetry install --with cuda -E cuda
+
+
 
 # Activate the shell
 poetry shell