From 680c3587d81badfc8df7ea411cc768e0f3e93e7b Mon Sep 17 00:00:00 2001 From: Richard Abrich Date: Sat, 13 Jan 2024 20:03:48 -0500 Subject: [PATCH] FORCE_CUDA=1 --- Dockerfile | 2 ++ 1 file changed, 2 insertions(+) diff --git a/Dockerfile b/Dockerfile index 49da107a..0fc55730 100644 --- a/Dockerfile +++ b/Dockerfile @@ -19,6 +19,8 @@ COPY . . # Install any needed packages specified in requirements.txt #RUN pip install --no-cache-dir -r requirements.txt +ENV FORCE_CUDA=1 + # Install additional dependencies from the README RUN pip install git+https://github.com/UX-Decoder/Segment-Everything-Everywhere-All-At-Once.git@package \ && pip install git+https://github.com/facebookresearch/segment-anything.git \