Spaces:
Runtime error
Runtime error
update import detectron2
Browse files- Dockerfile +11 -11
Dockerfile
CHANGED
@@ -20,7 +20,7 @@ RUN pip install 'git+https://github.com/facebookresearch/detectron2.git'
|
|
20 |
COPY --chown=user . /app
|
21 |
|
22 |
# Stage 2: CUDA and Detectron2
|
23 |
-
FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu18.04 as cuda
|
24 |
|
25 |
# Use noninteractive mode for apt-get
|
26 |
ENV DEBIAN_FRONTEND=noninteractive
|
@@ -47,21 +47,21 @@ COPY --from=base /app /app
|
|
47 |
|
48 |
# Install additional Python packages
|
49 |
RUN pip install --no-cache-dir scikit-build
|
50 |
-
RUN pip install --no-cache-dir --upgrade -r requirements.txt
|
51 |
RUN pip install --no-cache-dir tensorboard cmake onnx
|
52 |
RUN pip install --no-cache-dir 'git+https://github.com/facebookresearch/fvcore'
|
53 |
|
54 |
-
# Clone Detectron2 repository
|
55 |
-
RUN git clone https://github.com/facebookresearch/detectron2 detectron2_repo
|
56 |
|
57 |
-
# Set environment variables for Detectron2
|
58 |
-
ENV FORCE_CUDA="1"
|
59 |
-
ARG TORCH_CUDA_ARCH_LIST="Kepler;Kepler+Tesla;Maxwell;Maxwell+Tegra;Pascal;Volta;Turing"
|
60 |
-
ENV TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST}"
|
61 |
-
ENV FVCORE_CACHE="/tmp"
|
62 |
|
63 |
-
# Install Detectron2 in editable mode
|
64 |
-
RUN pip install --no-cache-dir -e detectron2_repo
|
65 |
|
66 |
# Set the working directory and environment
|
67 |
USER user
|
|
|
20 |
COPY --chown=user . /app
|
21 |
|
22 |
# Stage 2: CUDA and Detectron2
|
23 |
+
# FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu18.04 as cuda
|
24 |
|
25 |
# Use noninteractive mode for apt-get
|
26 |
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
47 |
|
48 |
# Install additional Python packages
|
49 |
RUN pip install --no-cache-dir scikit-build
|
50 |
+
# RUN pip install --no-cache-dir --upgrade -r requirements.txt
|
51 |
RUN pip install --no-cache-dir tensorboard cmake onnx
|
52 |
RUN pip install --no-cache-dir 'git+https://github.com/facebookresearch/fvcore'
|
53 |
|
54 |
+
# # Clone Detectron2 repository
|
55 |
+
# RUN git clone https://github.com/facebookresearch/detectron2 detectron2_repo
|
56 |
|
57 |
+
# # Set environment variables for Detectron2
|
58 |
+
# ENV FORCE_CUDA="1"
|
59 |
+
# ARG TORCH_CUDA_ARCH_LIST="Kepler;Kepler+Tesla;Maxwell;Maxwell+Tegra;Pascal;Volta;Turing"
|
60 |
+
# ENV TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST}"
|
61 |
+
# ENV FVCORE_CACHE="/tmp"
|
62 |
|
63 |
+
# # Install Detectron2 in editable mode
|
64 |
+
# RUN pip install --no-cache-dir -e detectron2_repo
|
65 |
|
66 |
# Set the working directory and environment
|
67 |
USER user
|