Hugging Face's logo

Spaces:
NN-BRD
/
MMpose
Build error

App Files Community
xmrt commited on
Commit
71e78c9
·
1 Parent(s): d94dadb

cuda

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -7,7 +7,8 @@ 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 --index-url https://download.pytorch.org/whl/cu117
 
11
  # --extra-index-url https://download.pytorch.org/whl/cu113
12
  RUN pip install --no-cache-dir --upgrade -U openmim
13
  RUN mim install --no-cache-dir --upgrade mmengine
 
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