chrisjcc commited on
Commit
1ef3684
·
verified ·
1 Parent(s): 03444d0

Check to see if there are GPU library installation available

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -4,6 +4,11 @@ from PIL import Image
4
  import base64
5
 
6
  import torch
 
 
 
 
 
7
  from transformers import pipeline
8
 
9
  from diffusers import EulerDiscreteScheduler
 
4
  import base64
5
 
6
  import torch
7
+ print(f"Is CUDA available: {torch.cuda.is_available()}")
8
+ # True
9
+ print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
10
+ # Tesla T4
11
+
12
  from transformers import pipeline
13
 
14
  from diffusers import EulerDiscreteScheduler