Library cudart is not initialized

[woaipichuli]

运行如下代码报错

from cpm_kernels import library
<module 'cpm_kernels.library.cudart' from '/usr/local/lib/python3.9/site-packages/cpm_kernels/library/cudart.py'>
library.cudart.cudaGetDevice()
Traceback (most recent call last):
File "", line 1, in
File "/usr/local/lib/python3.9/site-packages/cpm_kernels/library/base.py", line 72, in wrapper
raise RuntimeError("Library %s is not initialized" % self.__name)
RuntimeError: Library cudart is not initialized
exit()

采用docker容器启动，宿主机的驱动情况如下：

容器内安装了pytorch对应的cudatoolkit：
nvidia-cublas-cu11 11.10.3.66
nvidia-cuda-nvrtc-cu11 11.7.99
nvidia-cuda-runtime-cu11 11.7.99
nvidia-cudnn-cu11 8.5.0.96
尝试了将容器内的libcudart.so.11.0添加到环境变量LD_LIBRARY_PATH，但貌似并不是这个引起的报错