Commit
·
7298171
1
Parent(s):
f1c3907
Update Dockerfile
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
|
@@ -32,7 +32,8 @@ ENV HOME=/home/user \
|
|
| 32 |
# RUN conda install python=3.8
|
| 33 |
|
| 34 |
RUN pip install setuptools-rust
|
| 35 |
-
RUN pip install torch==1.11.0
|
|
|
|
| 36 |
RUN pip install gradio scikit-image pillow openmim
|
| 37 |
RUN pip install --upgrade setuptools==69.5.1
|
| 38 |
|
|
|
|
| 32 |
# RUN conda install python=3.8
|
| 33 |
|
| 34 |
RUN pip install setuptools-rust
|
| 35 |
+
RUN pip install torch==1.11.0 torchvision
|
| 36 |
+
# RUN pip install torch==1.11.0+cu115 torchvision==0.12.0+cu115 --extra-index-url https://download.pytorch.org/whl/cu115
|
| 37 |
RUN pip install gradio scikit-image pillow openmim
|
| 38 |
RUN pip install --upgrade setuptools==69.5.1
|
| 39 |
|