Commit
·
ef58dac
1
Parent(s):
bbd12c7
update Dockerfile
Browse files- Dockerfile +4 -4
Dockerfile
CHANGED
@@ -1,10 +1,6 @@
|
|
1 |
# Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
|
2 |
FROM nvcr.io/nvidia/pytorch:20.03-py3
|
3 |
|
4 |
-
# Install dependencies (pip or conda)
|
5 |
-
RUN pip install -U gsutil
|
6 |
-
# RUN pip install -U -r requirements.txt
|
7 |
-
|
8 |
# Create working directory
|
9 |
RUN mkdir -p /usr/src/app
|
10 |
WORKDIR /usr/src/app
|
@@ -12,6 +8,10 @@ WORKDIR /usr/src/app
|
|
12 |
# Copy contents
|
13 |
COPY . /usr/src/app
|
14 |
|
|
|
|
|
|
|
|
|
15 |
# Copy weights
|
16 |
#RUN python3 -c "from models import *; \
|
17 |
#attempt_download('weights/yolov5s.pt'); \
|
|
|
1 |
# Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
|
2 |
FROM nvcr.io/nvidia/pytorch:20.03-py3
|
3 |
|
|
|
|
|
|
|
|
|
4 |
# Create working directory
|
5 |
RUN mkdir -p /usr/src/app
|
6 |
WORKDIR /usr/src/app
|
|
|
8 |
# Copy contents
|
9 |
COPY . /usr/src/app
|
10 |
|
11 |
+
# Install dependencies (pip or conda)
|
12 |
+
#RUN pip install -r requirements.txt
|
13 |
+
RUN pip install -U gsutil
|
14 |
+
|
15 |
# Copy weights
|
16 |
#RUN python3 -c "from models import *; \
|
17 |
#attempt_download('weights/yolov5s.pt'); \
|