licensy commited on
Commit
13c30fa
·
verified ·
1 Parent(s): 53ff481

Fix CUDA: ctypes preload cuDNN + ldconfig + shutdown=14400

Browse files
Files changed (1) hide show
  1. miner.py +5 -1
miner.py CHANGED
@@ -3,11 +3,15 @@ import math
3
  import os
4
  import glob
5
  import site
 
6
 
7
- # Ensure pip-installed NVIDIA CUDA/cuDNN libraries are discoverable
8
  for sp in site.getsitepackages():
9
  for d in glob.glob(os.path.join(sp, 'nvidia', '*', 'lib')):
10
  os.environ['LD_LIBRARY_PATH'] = d + ':' + os.environ.get('LD_LIBRARY_PATH', '')
 
 
 
11
 
12
  import cv2
13
  import numpy as np
 
3
  import os
4
  import glob
5
  import site
6
+ import ctypes
7
 
8
+ # Preload pip-installed NVIDIA cuDNN so onnxruntime can use CUDAExecutionProvider
9
  for sp in site.getsitepackages():
10
  for d in glob.glob(os.path.join(sp, 'nvidia', '*', 'lib')):
11
  os.environ['LD_LIBRARY_PATH'] = d + ':' + os.environ.get('LD_LIBRARY_PATH', '')
12
+ _cudnn = os.path.join(d, 'libcudnn.so.9')
13
+ if os.path.exists(_cudnn):
14
+ ctypes.CDLL(_cudnn, mode=ctypes.RTLD_GLOBAL)
15
 
16
  import cv2
17
  import numpy as np