xx12

Paused

Nick088 commited on Apr 1

Commit

fb6f00b

•

1 Parent(s): 5f091f6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import os
 import torch
 if torch.cuda.is_available():
-  !apt-get install nvidia-cuda-toolkit
   print("Using GPU")
 else:
   print("Using CPU")

 import torch
 if torch.cuda.is_available():
+  subprocess.run(["apt-get", "install", "nvidia-cuda-toolkit"], check=True)
   print("Using GPU")
 else:
   print("Using CPU")