reidddd commited on
Commit
9506fa0
·
1 Parent(s): 3dfc91d

update import detectron2

Browse files
Files changed (1) hide show
  1. 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