Spaces:

julien-c
/

zero-gpu-demo

Sleeping

julien-c HF Staff commited on Nov 24, 2023

Commit

73eea05

verified ·

1 Parent(s): 3bf1824

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,9 +12,11 @@ CUSTOM_CSS = """
 """
 zero = torch.Tensor([0]).cuda()
 @spaces.GPU
 def run_gpu() -> str:
     output: str = ""
     try:
         output = subprocess.check_output(["nvidia-smi"], text=True)

 """
 zero = torch.Tensor([0]).cuda()
+print(zero.device) # <-- 'cpu' 🤔
 @spaces.GPU
 def run_gpu() -> str:
+    print(zero.device) # <-- 'cuda:0' 🤗
     output: str = ""
     try:
         output = subprocess.check_output(["nvidia-smi"], text=True)