diff --git a/Dockerfile b/Dockerfile index e29ddca..d24039d 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,6 +1,9 @@ # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch FROM nvcr.io/nvidia/pytorch:20.03-py3 -RUN pip install -U gsutil + +# Install dependencies +COPY requirements.txt . +RUN pip install -r requirements.txt gsutil # Create working directory RUN mkdir -p /usr/src/app @@ -9,9 +12,6 @@ WORKDIR /usr/src/app # Copy contents COPY . /usr/src/app -# Install dependencies (pip or conda) -#RUN pip install -r requirements.txt - # Copy weights #RUN python3 -c "from models import *; \ #attempt_download('weights/yolov5s.pt'); \