Paolo-Fraccaro commited on
Commit
246d612
·
1 Parent(s): 982fd4b
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -58,7 +58,7 @@ RUN conda install pillow -y
58
  RUN conda install -c conda-forge pytorch-cpu==1.7.1
59
 
60
  # RUN conda install -c pytorch torchvision-cpu==0.8.2
61
- RUN pip install torchvision==0.8.2+cpu
62
 
63
  RUN pip install openmin
64
 
 
58
  RUN conda install -c conda-forge pytorch-cpu==1.7.1
59
 
60
  # RUN conda install -c pytorch torchvision-cpu==0.8.2
61
+ RUN pip install torchvision-cpu==0.8.2
62
 
63
  RUN pip install openmin
64