FORCE_CUDA=1

OpenAdaptAI · Jan 14, 2024 · 680c358 · 680c358
1 parent 5e02317
commit 680c358
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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 \