update

app.py

@@ -39,7 +39,7 @@ os.makedirs(opts.save_dir,exist_ok=True)
 test_tensor = torch.Tensor([0]).cuda()
 opts.device = str(test_tensor.device)
 os.system('pip install iopath')
-os.system("pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'")
+os.system("FORCE_CUDA=1 TORCH_CUDA_ARCH_LIST='6.0;6.1;7.0;7.5;8.0;8.6+PTX' CUDA_HOME='/usr/local/cuda' pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'")
 from viewcrafter import ViewCrafter
 
 def viewcrafter_demo(opts):