ljy266987 commited on
Commit
659c3bd
·
1 Parent(s): 852b612
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -30,6 +30,10 @@ MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
30
  ##
31
  # 检查CUDA是否可用
32
  def print_gpu():
 
 
 
 
33
  if torch.cuda.is_available():
34
  print("CUDA is available. Listing available GPUs:")
35
  # 获取并打印GPU数量
 
30
  ##
31
  # 检查CUDA是否可用
32
  def print_gpu():
33
+ result = subprocess.run(
34
+ ['nvidia-smi'], text=True
35
+ )
36
+
37
  if torch.cuda.is_available():
38
  print("CUDA is available. Listing available GPUs:")
39
  # 获取并打印GPU数量