kyleleey commited on
Commit
e14627f
1 Parent(s): 605aec0
Files changed (2) hide show
  1. Dockerfile +2 -0
  2. app.py +2 -2
Dockerfile CHANGED
@@ -57,6 +57,8 @@ RUN git clone -b v0.7.4 https://github.com/facebookresearch/pytorch3d.git /home/
57
  WORKDIR /home/user/pytorch3d-0.7.4
58
  RUN python setup.py install --user
59
 
 
 
60
  # Set the working directory to the user's home directory
61
  WORKDIR $HOME/app
62
 
 
57
  WORKDIR /home/user/pytorch3d-0.7.4
58
  RUN python setup.py install --user
59
 
60
+ RUN pip install git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch
61
+
62
  # Set the working directory to the user's home directory
63
  WORKDIR $HOME/app
64
 
app.py CHANGED
@@ -7,8 +7,8 @@ import argparse
7
  import sys
8
  import torch
9
 
10
- if os.getenv('SYSTEM') == 'spaces':
11
- os.system('pip install --global-option="--no-networks" git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch')
12
  # os.system('pip install fvcore iopath')
13
  # os.system('pip install "git+https://github.com/facebookresearch/pytorch3d.git"')
14
 
 
7
  import sys
8
  import torch
9
 
10
+ # if os.getenv('SYSTEM') == 'spaces':
11
+ # os.system('pip install --global-option="--no-networks" git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch')
12
  # os.system('pip install fvcore iopath')
13
  # os.system('pip install "git+https://github.com/facebookresearch/pytorch3d.git"')
14