Spaces:

Doubiiu
/

ViewCrafter

Running on Zero

Drexubery commited on Sep 3

Commit

30ace45

•

1 Parent(s): 6679c1c

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,7 +38,7 @@ opts.save_dir = './'
 os.makedirs(opts.save_dir,exist_ok=True)
 test_tensor = torch.Tensor([0]).cuda()
 opts.device = str(test_tensor.device)
-os.system("pip install 'git+https://github.com/facebookresearch/pytorch3d.git'")
 from viewcrafter import ViewCrafter
 def viewcrafter_demo(opts):

 os.makedirs(opts.save_dir,exist_ok=True)
 test_tensor = torch.Tensor([0]).cuda()
 opts.device = str(test_tensor.device)
+os.system("FORCE_CUDA=1 pip install 'git+https://github.com/facebookresearch/pytorch3d.git'")
 from viewcrafter import ViewCrafter
 def viewcrafter_demo(opts):