carlosgomes98
commited on
Commit
•
7225b33
1
Parent(s):
d4fb1df
change cuda version
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
@@ -32,7 +32,7 @@ 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
|
38 |
|
@@ -48,7 +48,7 @@ RUN git checkout 9968269915db8402bf4a6d0549df9df57d489e5a
|
|
48 |
|
49 |
RUN pip install -e .
|
50 |
|
51 |
-
RUN mim install mmcv-full==1.6.2 -f https://download.openmmlab.com/mmcv/dist/11.
|
52 |
|
53 |
# Set the working directory to the user's home directory
|
54 |
WORKDIR $HOME/app
|
|
|
32 |
# RUN conda install python=3.8
|
33 |
|
34 |
RUN pip install setuptools-rust
|
35 |
+
RUN pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 --extra-index-url https://download.pytorch.org/whl/cu113
|
36 |
RUN pip install gradio scikit-image pillow openmim
|
37 |
RUN pip install --upgrade setuptools
|
38 |
|
|
|
48 |
|
49 |
RUN pip install -e .
|
50 |
|
51 |
+
RUN mim install mmcv-full==1.6.2 -f https://download.openmmlab.com/mmcv/dist/11.3/1.11.0/index.html
|
52 |
|
53 |
# Set the working directory to the user's home directory
|
54 |
WORKDIR $HOME/app
|