|
|
@@ -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'); \ |