Commit ·
671595c
1
Parent(s): 7732433
debug LD_LIBRARY_PATH
Browse files- handler.py +5 -4
handler.py
CHANGED
|
@@ -8,6 +8,7 @@ import torch
|
|
| 8 |
# stdout, stderr = process.communicate()
|
| 9 |
|
| 10 |
from faster_whisper import WhisperModel
|
|
|
|
| 11 |
import os
|
| 12 |
import nvidia.cublas.lib
|
| 13 |
import nvidia.cudnn.lib
|
|
@@ -26,10 +27,10 @@ import logging
|
|
| 26 |
|
| 27 |
logger = logging.getLogger(__name__)
|
| 28 |
|
| 29 |
-
logger.info(
|
| 30 |
-
logger.info(
|
| 31 |
-
os.environ["LD_LIBRARY_PATH"] = (
|
| 32 |
-
|
| 33 |
)
|
| 34 |
SAMPLE_RATE = 16000
|
| 35 |
|
|
|
|
| 8 |
# stdout, stderr = process.communicate()
|
| 9 |
|
| 10 |
from faster_whisper import WhisperModel
|
| 11 |
+
from pathlib import Path
|
| 12 |
import os
|
| 13 |
import nvidia.cublas.lib
|
| 14 |
import nvidia.cudnn.lib
|
|
|
|
| 27 |
|
| 28 |
logger = logging.getLogger(__name__)
|
| 29 |
|
| 30 |
+
logger.info(Path(nvidia.cublas.lib.__file__).parent)
|
| 31 |
+
logger.info(Path(nvidia.cudnn.lib.__file__).parent)
|
| 32 |
+
os.environ["LD_LIBRARY_PATH"] = ":".join(
|
| 33 |
+
[Path(nvidia.cublas.lib.__file__).parent, Path(nvidia.cudnn.lib.__file__).parent]
|
| 34 |
)
|
| 35 |
SAMPLE_RATE = 16000
|
| 36 |
|