Fix cuda images

Author: OskarLiew

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

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

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