File tree Expand file tree Collapse file tree 2 files changed +3
-5
lines changed Expand file tree Collapse file tree 2 files changed +3
-5
lines changed Original file line number Diff line number Diff line change 1
- ARG CUDA_VERSION=12.1.1-cudnn8-runtime-ubuntu20.04
2
1
ARG PYTHON_VERSION=3.8
3
2
ARG POETRY_VERSION=1.7.1
4
- FROM nvidia/cuda:$CUDA_VERSION
3
+ FROM ubuntu:20.04
5
4
ARG PYTHON_VERSION
6
5
ARG POETRY_VERSION
7
6
WORKDIR /app
Original file line number Diff line number Diff line change 1
- ARG CUDA_VERSION=12.1.1-cudnn8-runtime-ubuntu20.04
2
1
ARG PYTHON_VERSION=3.8
3
2
ARG POETRY_VERSION=1.7.1
4
3
@@ -23,7 +22,7 @@ RUN poetry export -E eflomal --without-hashes -f requirements.txt > requirements
23
22
COPY . /src
24
23
RUN poetry build
25
24
26
- FROM nvidia/cuda:$CUDA_VERSION
25
+ FROM ubuntu:20.04
27
26
28
27
ARG PYTHON_VERSION=3.8
29
28
@@ -58,7 +57,7 @@ RUN ln -sfn /usr/bin/python${PYTHON_VERSION} /usr/bin/python3 & \
58
57
59
58
# Install dependencies from poetry
60
59
COPY --from=builder /src/requirements.txt .
61
- RUN pip install --no-cache-dir - r requirements.txt && rm requirements.txt
60
+ RUN pip install -r requirements.txt && rm requirements.txt
62
61
63
62
# Set eflomal path
64
63
ENV EFLOMAL_PATH=/usr/local/lib/python3.8/dist-packages/eflomal/bin
You can’t perform that action at this time.
0 commit comments