From 53ecaec28bdbc4dda7c0703049ce9fd98fab422e Mon Sep 17 00:00:00 2001 From: mshannon-sil Date: Mon, 30 Sep 2024 18:50:00 +0000 Subject: [PATCH] use non cuda ubuntu docker image --- .devcontainer/Dockerfile | 3 +-- Dockerfile | 5 ++--- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile index 36a73aae..0d66aee0 100644 --- a/.devcontainer/Dockerfile +++ b/.devcontainer/Dockerfile @@ -1,7 +1,6 @@ -ARG CUDA_VERSION=12.1.1-cudnn8-runtime-ubuntu20.04 ARG PYTHON_VERSION=3.8 ARG POETRY_VERSION=1.7.1 -FROM nvidia/cuda:$CUDA_VERSION +FROM ubuntu:20.04 ARG PYTHON_VERSION ARG POETRY_VERSION WORKDIR /app diff --git a/Dockerfile b/Dockerfile index 41059f08..c1f83d93 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,3 @@ -ARG CUDA_VERSION=12.1.1-cudnn8-runtime-ubuntu20.04 ARG PYTHON_VERSION=3.8 ARG POETRY_VERSION=1.7.1 @@ -23,7 +22,7 @@ RUN poetry export -E eflomal --without-hashes -f requirements.txt > requirements COPY . /src RUN poetry build -FROM nvidia/cuda:$CUDA_VERSION +FROM ubuntu:20.04 ARG PYTHON_VERSION=3.8 @@ -58,7 +57,7 @@ RUN ln -sfn /usr/bin/python${PYTHON_VERSION} /usr/bin/python3 & \ # Install dependencies from poetry COPY --from=builder /src/requirements.txt . -RUN pip install --no-cache-dir -r requirements.txt && rm requirements.txt +RUN pip install -r requirements.txt && rm requirements.txt # Set eflomal path ENV EFLOMAL_PATH=/usr/local/lib/python3.8/dist-packages/eflomal/bin