ericmattmann commited on
Commit
671595c
·
1 Parent(s): 7732433

debug LD_LIBRARY_PATH

Browse files
Files changed (1) hide show
  1. 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(os.path.dirname(nvidia.cublas.lib.__file__))
30
- logger.info(os.path.dirname(nvidia.cudnn.lib.__file__))
31
- os.environ["LD_LIBRARY_PATH"] = (
32
- os.path.dirname(nvidia.cublas.lib.__file__) + ":" + os.path.dirname(nvidia.cudnn.lib.__file__)
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