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