Update gradio_app.py

gradio_app.py

@@ -17,6 +17,11 @@ logger = logging.getLogger(__name__)
 logging.basicConfig(level=logging.DEBUG)
 
 # Ensure necessary devices are available
+print("torch.__version__: ", torch.__version__)
+import subprocess
+
+cuda_version = subprocess.check_output(['nvcc', '--version']).decode('utf-8')
+print("cuda_version: ", cuda_version)
 device = "cuda" if torch.cuda.is_available() else "cpu"
 logger.info("device: {}".format(device))