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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -35
app.py CHANGED
@@ -1,38 +1,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
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import os
2
  import io
3