提交 ef58dac9 authored 作者: Glenn Jocher's avatar Glenn Jocher

update Dockerfile

上级 bbd12c7c
# 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'); \
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论