Commit
·
982fd4b
1
Parent(s):
d9c040b
fix torchvision
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
|
@@ -57,7 +57,8 @@ RUN conda install pillow -y
|
|
| 57 |
|
| 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 |
|
| 62 |
RUN pip install openmin
|
| 63 |
|
|
|
|
| 57 |
|
| 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 |
|