Hugging Face's logo

Spaces:
NN-BRD
/
MMpose
Build error

App Files Community
xmrt commited on
Commit
58992fb
·
1 Parent(s): c227693

cuda

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -4,7 +4,7 @@ WORKDIR /code
4
  COPY ./requirements.txt /code/requirements.txt
5
 
6
  #RUN pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.ht
7
- RUN pip install torch==1.9.1 torchvision==0.10.1 -f https://download.pytorch.org/whl/cu113
8
  #https://download.pytorch.org/whl/torch_stable.ht
9
  RUN pip install --no-cache-dir --upgrade -U openmim
10
  RUN mim install --no-cache-dir --upgrade mmengine
 
4
  COPY ./requirements.txt /code/requirements.txt
5
 
6
  #RUN pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.ht
7
+ RUN pip install torch==1.9.1 torchvision==0.10.1 -f https://download.pytorch.org/whl/cu111
8
  #https://download.pytorch.org/whl/torch_stable.ht
9
  RUN pip install --no-cache-dir --upgrade -U openmim
10
  RUN mim install --no-cache-dir --upgrade mmengine