yinwentao commited on
Commit
fa0103d
1 Parent(s): 09f6ae8

updateRequirements

Browse files
Files changed (2) hide show
  1. Dockerfile +5 -2
  2. app.py +1 -1
Dockerfile CHANGED
@@ -39,10 +39,13 @@ ENV HOME=/home/user \
39
 
40
  RUN pip3 install --no-cache-dir -U torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
41
  RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
42
-
 
 
 
 
43
  # Set the working directory to the user's home directory
44
  WORKDIR $HOME/app
45
- RUN ls -la /home
46
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
47
  COPY --chown=user . $HOME/app
48
 
 
39
 
40
  RUN pip3 install --no-cache-dir -U torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
41
  RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
42
+ RUN pip3 install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu113_pyt1121/download.html
43
+ RUN pip3 install tensorflow-gpu==2.8.1
44
+ RUN pip3 install ./shencoder
45
+ RUN pip3 install ./gridencoder
46
+ RUN pip3 install ./raymarching
47
  # Set the working directory to the user's home directory
48
  WORKDIR $HOME/app
 
49
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
50
  COPY --chown=user . $HOME/app
51
 
app.py CHANGED
@@ -1,7 +1,7 @@
1
 
2
  import os
3
  from huggingface_hub import HfFileSystem
4
- fs = HfFileSystem()
5
 
6
  # os.system(f"pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu113_pyt1121/download.html")
7
  # os.system(f"pip install tensorflow-gpu==2.8.1")
 
1
 
2
  import os
3
  from huggingface_hub import HfFileSystem
4
+
5
 
6
  # os.system(f"pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu113_pyt1121/download.html")
7
  # os.system(f"pip install tensorflow-gpu==2.8.1")