Paolo-Fraccaro commited on
Commit
ef216a2
·
1 Parent(s): 48a0f03
Files changed (1) hide show
  1. Dockerfile +1 -3
Dockerfile CHANGED
@@ -55,9 +55,7 @@ RUN conda install pillow -y
55
 
56
  # RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
57
 
58
- RUN conda install -c conda-forge pytorch-cpu==1.7.1
59
-
60
- RUN conda install -c conda-forge torchvision==0.8.2
61
  # RUN pip install torchvision-cpu==0.8.2
62
 
63
  RUN pip install openmin
 
55
 
56
  # RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
57
 
58
+ RUN conda install -c pytorch pytorch==1.7.1 torchvision==0.8.2
 
 
59
  # RUN pip install torchvision-cpu==0.8.2
60
 
61
  RUN pip install openmin