Doubiiu commited on
Commit
579479d
1 Parent(s): 90d9c11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
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):