fffiloni commited on
Commit
36dbf31
1 Parent(s): ad73e3e

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
2
 
3
  ENV DEBIAN_FRONTEND=noninteractive
4
 
@@ -33,7 +33,7 @@ RUN pip install -q pytorch-lightning omegaconf einops yacs trimesh mediapipe rtr
33
  RUN pip install -q git+https://gitlab.eecs.umich.edu/ngv-python-modules/opendr
34
  RUN pip install -q https://download.pytorch.org/whl/cu121/xformers-0.0.22.post7-cp310-cp310-manylinux2014_x86_64.whl
35
 
36
- RUN pip install --no-cache-dir diffusers gradio opencv-python transformers
37
 
38
  RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/HandRefiner/resolve/main/basicModel_neutral_lbs_10_207_0_v1.0.0.pkl -d $HOME/app/MeshGraphormer/src/modeling/data -o basicModel_neutral_lbs_10_207_0_v1.0.0.pkl
39
  RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/HandRefiner/resolve/main/MANO_RIGHT.pkl -d $HOME/app/MeshGraphormer/src/modeling/data -o MANO_RIGHT.pkl
 
1
+ FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime
2
 
3
  ENV DEBIAN_FRONTEND=noninteractive
4
 
 
33
  RUN pip install -q git+https://gitlab.eecs.umich.edu/ngv-python-modules/opendr
34
  RUN pip install -q https://download.pytorch.org/whl/cu121/xformers-0.0.22.post7-cp310-cp310-manylinux2014_x86_64.whl
35
 
36
+ RUN pip install --no-cache-dir -r -r requirements.txt gradio
37
 
38
  RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/HandRefiner/resolve/main/basicModel_neutral_lbs_10_207_0_v1.0.0.pkl -d $HOME/app/MeshGraphormer/src/modeling/data -o basicModel_neutral_lbs_10_207_0_v1.0.0.pkl
39
  RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/HandRefiner/resolve/main/MANO_RIGHT.pkl -d $HOME/app/MeshGraphormer/src/modeling/data -o MANO_RIGHT.pkl