Hugging Face's logo

Spaces:
NN-BRD
/
MMpose
Build error

App Files Community
xmrt commited on
Commit
e128b40
•
1 Parent(s): 71e78c9

cuda

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -7,9 +7,10 @@ COPY ./requirements.txt /code/requirements.txt
7
  #RUN pip install torch==1.9.1 torchvision==0.10.1 -f https://download.pytorch.org/whl/torch_stable.ht -extra-index-url https://download.pytorch.org/whl/cu111
8
  #https://download.pytorch.org/whl/torch_stable.ht
9
 
10
- RUN pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu113
11
  # --index-url https://download.pytorch.org/whl/cu117
12
  # --extra-index-url https://download.pytorch.org/whl/cu113
 
13
  RUN pip install --no-cache-dir --upgrade -U openmim
14
  RUN mim install --no-cache-dir --upgrade mmengine
15
  RUN mim install "mmcv>=2.0.1"
 
7
  #RUN pip install torch==1.9.1 torchvision==0.10.1 -f https://download.pytorch.org/whl/torch_stable.ht -extra-index-url https://download.pytorch.org/whl/cu111
8
  #https://download.pytorch.org/whl/torch_stable.ht
9
 
10
+ RUN pip install torch --extra-index-url https://download.pytorch.org/whl/cu113
11
  # --index-url https://download.pytorch.org/whl/cu117
12
  # --extra-index-url https://download.pytorch.org/whl/cu113
13
+ RUN pip install torchvision --extra-index-url https://download.pytorch.org/whl/cu113
14
  RUN pip install --no-cache-dir --upgrade -U openmim
15
  RUN mim install --no-cache-dir --upgrade mmengine
16
  RUN mim install "mmcv>=2.0.1"