Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -41,6 +41,11 @@ opts.device = str(test_tensor.device)
|
|
41 |
# spaces.GPU(os.system("FORCE_CUDA=1 pip install 'git+https://github.com/facebookresearch/pytorch3d.git'"))
|
42 |
os.system("mkdir -p checkpoints/ && wget https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth -P checkpoints/")
|
43 |
|
|
|
|
|
|
|
|
|
|
|
44 |
from viewcrafter import ViewCrafter
|
45 |
|
46 |
def viewcrafter_demo(opts):
|
|
|
41 |
# spaces.GPU(os.system("FORCE_CUDA=1 pip install 'git+https://github.com/facebookresearch/pytorch3d.git'"))
|
42 |
os.system("mkdir -p checkpoints/ && wget https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth -P checkpoints/")
|
43 |
|
44 |
+
print(f'>>> System info: CUDA: {torch.version.cuda}')
|
45 |
+
os.system('ls -l /usr/local | grep cuda')
|
46 |
+
os.system('nvcc -V')
|
47 |
+
os.system('nvidia-smi')
|
48 |
+
|
49 |
from viewcrafter import ViewCrafter
|
50 |
|
51 |
def viewcrafter_demo(opts):
|