update

app.py

@@ -3,6 +3,10 @@ import gradio as gr
 import os
 import subprocess
 
+import torch 
+print(torch.__version__)
+print(torch.version.cuda)
+
 def install_cuda_toolkit():
     # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
     CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
@@ -25,7 +29,6 @@ install_cuda_toolkit()
 
 os.system('pip install ./thirdparty/DROID-SLAM')
 os.system('pip install ./thirdparty/DROID-SLAM/thirdparty/lietorch')
-import torch
 import numpy as np
 from easydict import EasyDict
 from scripts.scripts_test_video.detect_track_video import detect_track_video