Skip to content

Commit

Permalink
PyTorch 1.6.0 update with native AMP (ultralytics#573)
Browse files Browse the repository at this point in the history
  • Loading branch information
glenn-jocher authored Jul 31, 2020
1 parent 0dd5f50 commit 678c4be
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -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
Expand All @@ -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'); \
Expand Down

0 comments on commit 678c4be

Please sign in to comment.