chanfee commited on
Commit
49d77ef
·
verified ·
1 Parent(s): 468ff7c

Inspect cuda issuse...

Files changed (1) hide show
  1. app.py +35 -0
app.py CHANGED
@@ -1,3 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import os
2
  import io
3
 
 
1
+ #DEBUG
2
+ import os
3
+ import sys
4
+ import torch
5
+ import transformers
6
+ import gradio
7
+ import platform
8
+ import subprocess
9
+
10
+ print("Python Version:", sys.version)
11
+ print("Platform:", platform.platform())
12
+ print("Torch Version:", torch.__version__)
13
+ print("Transformers Version:", transformers.__version__)
14
+ print("Gradio Version:", gradio.__version__)
15
+
16
+ print("CUDA Available:", torch.cuda.is_available())
17
+ if torch.cuda.is_available():
18
+ print("CUDA Device Name:", torch.cuda.get_device_name(0))
19
+ print("CUDA Version:", torch.version.cuda)
20
+ print("CUDNN Version:", torch.backends.cudnn.version())
21
+ print("CUDA Memory Allocated:", torch.cuda.memory_allocated(0))
22
+ print("CUDA Memory Cached:", torch.cuda.memory_reserved(0))
23
+
24
+ print("Environment Variables:")
25
+ for key, value in os.environ.items():
26
+ print(f"{key}: {value}")
27
+
28
+ # Print NVIDIA GPU details
29
+ try:
30
+ subprocess.run(["nvidia-smi"], check=True)
31
+ except Exception as e:
32
+ print("nvidia-smi failed:", e)
33
+
34
+
35
+
36
  import os
37
  import io
38