yinwentao
commited on
Commit
·
fa0103d
1
Parent(s):
09f6ae8
updateRequirements
Browse files- Dockerfile +5 -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 |
-
|
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")
|