ReubenSun commited on
Commit
d18d029
·
1 Parent(s): d11e8f6
Files changed (2) hide show
  1. app.py +2 -0
  2. requirements.txt +0 -1
app.py CHANGED
@@ -21,6 +21,8 @@ os.system("pip list | grep torch")
21
  os.system('nvcc -V')
22
  print("cd /home/user/app/step1x3d_texture/differentiable_renderer/ && python setup.py install")
23
  os.system("cd /home/user/app/step1x3d_texture/differentiable_renderer/ && python setup.py install")
 
 
24
 
25
  subprocess.run(shlex.split("pip install custom_rasterizer-0.1-cp310-cp310-linux_x86_64.whl"), check=True)
26
  import time
 
21
  os.system('nvcc -V')
22
  print("cd /home/user/app/step1x3d_texture/differentiable_renderer/ && python setup.py install")
23
  os.system("cd /home/user/app/step1x3d_texture/differentiable_renderer/ && python setup.py install")
24
+ print('FORCE_CUDA=1 pip install "git+https://github.com/facebookresearch/pytorch3d.git@stable')
25
+ os.system('FORCE_CUDA=1 pip install "git+https://github.com/facebookresearch/pytorch3d.git@stable')
26
 
27
  subprocess.run(shlex.split("pip install custom_rasterizer-0.1-cp310-cp310-linux_x86_64.whl"), check=True)
28
  import time
requirements.txt CHANGED
@@ -49,7 +49,6 @@ pygltflib==1.16.4
49
  kornia==0.8.0
50
  cupy-cuda12x==13.4.1
51
  git+https://github.com/NVlabs/nvdiffrast.git
52
- git+https://github.com/facebookresearch/pytorch3d.git@stable
53
  torch==2.5.1
54
  torchvision==0.20.1
55
  torchaudio==2.5.1
 
49
  kornia==0.8.0
50
  cupy-cuda12x==13.4.1
51
  git+https://github.com/NVlabs/nvdiffrast.git
 
52
  torch==2.5.1
53
  torchvision==0.20.1
54
  torchaudio==2.5.1