.

app.py

@@ -210,7 +210,8 @@ with gr.Blocks() as demo:
     # JavaScriptの読み込み
     demo.load(js=script)
 
-# nvidia-smiをprint
-print(os.popen("nvidia-smi").read())
+# cudaのバージョンをprint
+import torch
+print("CUDA Version:", torch.version.cuda)
 
 demo.launch()