Spaces:
Runtime error
Runtime error
liuyang
commited on
Commit
·
b5b0753
1
Parent(s):
d3ed5e3
cuda check
Browse files
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 |
-
|
| 56 |
-
|
| 57 |
-
"cuDNN
|
|
|
|
|
|
|
| 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
|