update Dockerfile
This commit is contained in:
1 parent
bbd12c7cfe
commit
ef58dac963
1 file changed
+4
-4
+4
-4
@@ -1,10 +1,6 @@
|
|||||||
# 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
|
||||||
@@ -12,6 +8,10 @@ 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'); \
|
||||||
|
|||||||
Reference in new issue
Block a user