carlosgomes98 commited on
Commit
7225b33
1 Parent(s): d4fb1df

change cuda version

Browse files
Files changed (1) hide show
  1. 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+cu115 torchvision==0.12.0+cu115 --extra-index-url https://download.pytorch.org/whl/cu115
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.5/1.11.0/index.html
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