From 8bf3cff8755a78ef7b1cec104b5907f3544a0c63 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 31 Jul 2020 11:10:03 -0700 Subject: [PATCH] PyTorch 1.6.0 update with native AMP (#573) --- Dockerfile | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) 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'); \