Hugging Face's logo

Spaces:
NN-BRD
/
MMpose
Build error

App Files Community
xmrt commited on
Commit
cfe22ee
•
1 Parent(s): 26a664d

cuda

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. requirements.txt +3 -2
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/torch_stable.ht -extra-index-url 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
 
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/torch_stable.ht -extra-index-url 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
requirements.txt CHANGED
@@ -3,5 +3,6 @@ numpy
3
  opencv-python
4
  ultralytics
5
  lap
6
- torchvision
7
- MoviePy
 
 
3
  opencv-python
4
  ultralytics
5
  lap
6
+ MoviePy
7
+ --extra-index-url https://download.pytorch.org/whl/cu113
8
+ torch