Docker / Dockerfile
SalmanAboAraj's picture
Create Dockerfile
42fc073 verified
raw
history blame contribute delete
537 Bytes
FROM python:3.10
WORKDIR /app
# Step 1: Install PyTorch
RUN pip install --no-cache-dir --find-links https://download.pytorch.org/whl/torch_stable.html \
torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1
# Step 2: Install Detectron2 without build isolation
RUN pip install --no-cache-dir --no-build-isolation \
git+https://github.com/facebookresearch/detectron2.git
# Step 3: Install other requirements
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
COPY . .
CMD ["python", "app.py"]