wenxiang guo commited on
Commit
99f9a99
·
verified ·
1 Parent(s): 3af9bd5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import torch
2
  print(f"Is CUDA available: {torch.cuda.is_available()}")
3
- print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
4
  import spaces
5
  import argparse, os, sys, glob
6
  import pathlib
 
1
  import torch
2
  print(f"Is CUDA available: {torch.cuda.is_available()}")
3
+ # print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
4
  import spaces
5
  import argparse, os, sys, glob
6
  import pathlib