sandbox338 commited on
Commit
d211dc2
·
verified ·
1 Parent(s): 8ea9252

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -0
Dockerfile CHANGED
@@ -20,6 +20,9 @@ COPY src/ ./src/
20
  # Install the Python dependencies
21
  RUN pip3 install -r requirements.txt
22
 
 
 
 
23
  # Install torchvision
24
  RUN pip3 install torchvision==0.15.2
25
 
 
20
  # Install the Python dependencies
21
  RUN pip3 install -r requirements.txt
22
 
23
+ RUN pip3 install numpy
24
+
25
+
26
  # Install torchvision
27
  RUN pip3 install torchvision==0.15.2
28