Browse Source

update Dockerfile

5.0
Glenn Jocher 4 years ago
parent
commit
ef58dac963
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      Dockerfile

+ 4
- 4
Dockerfile View File

# Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
FROM nvcr.io/nvidia/pytorch:20.03-py3 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 # Create working directory
RUN mkdir -p /usr/src/app RUN mkdir -p /usr/src/app
WORKDIR /usr/src/app WORKDIR /usr/src/app
# Copy contents # Copy contents
COPY . /usr/src/app COPY . /usr/src/app


# Install dependencies (pip or conda)
#RUN pip install -r requirements.txt
RUN pip install -U gsutil

# Copy weights # Copy weights
#RUN python3 -c "from models import *; \ #RUN python3 -c "from models import *; \
#attempt_download('weights/yolov5s.pt'); \ #attempt_download('weights/yolov5s.pt'); \

Loading…
Cancel
Save