measmonysuon commited on
Commit
ca60ac1
1 Parent(s): cd028fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -30,6 +30,7 @@ client_image = Client("measmonysuon/DALLE-4K")
30
  user_info = whoami(token=API_TOKEN)
31
  print("User Info:", user_info)
32
 
 
33
  print(f"Is CUDA available: {torch.cuda.is_available()}")
34
  # True
35
  print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
 
30
  user_info = whoami(token=API_TOKEN)
31
  print("User Info:", user_info)
32
 
33
+ print(torch.version.cuda)
34
  print(f"Is CUDA available: {torch.cuda.is_available()}")
35
  # True
36
  print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")