liuyang commited on
Commit
b5b0753
·
1 Parent(s): d3ed5e3

cuda check

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -51,10 +51,12 @@ if _cudnn:
51
  os.environ["LD_LIBRARY_PATH"] = _cudnn + ":" + os.environ.get("LD_LIBRARY_PATH", "")
52
  # -------------------------------------------------------------
53
 
54
- import torch
55
- print("CUDA:", torch.cuda.is_available(),
56
- "cuDNN avail:", torch.backends.cudnn.is_available(),
57
- "cuDNN ver:", torch.backends.cudnn.version())
 
 
58
 
59
  import gradio as gr
60
  import torchaudio
 
51
  os.environ["LD_LIBRARY_PATH"] = _cudnn + ":" + os.environ.get("LD_LIBRARY_PATH", "")
52
  # -------------------------------------------------------------
53
 
54
+ import torch, ctranslate2, os
55
+
56
+ print("torch", torch.__version__, "CUDA build:", torch.version.cuda,
57
+ "cuDNN:", torch.backends.cudnn.version())
58
+ print("CT2:", ctranslate2.__version__)
59
+ print("LD_LIBRARY_PATH has cudnn/lib?", any("cudnn/lib" in p for p in os.environ.get("LD_LIBRARY_PATH","").split(":")))
60
 
61
  import gradio as gr
62
  import torchaudio