OSError undefined symbol when load deepmd.infer

[plumbum082]

Hi ~
I installed python interface on Bohrium with this command "pip install deepmd-kit[gpu,cu11,ipi], then I ran the .py file with this line "from deepmd.infer import DeepPot" , but it gave me this error,
Traceback (most recent call last):
File "/data/dp_cmd/./test_ene.py", line 31, in
from deepmd.infer import DeepPot
File "/opt/mamba/lib/python3.10/site-packages/deepmd/init.py", line 10, in
import deepmd.utils.network as network
File "/opt/mamba/lib/python3.10/site-packages/deepmd/utils/init.py", line 2, in
from .data import (
File "/opt/mamba/lib/python3.10/site-packages/deepmd/utils/data.py", line 11, in
from deepmd.env import (
File "/opt/mamba/lib/python3.10/site-packages/deepmd/env.py", line 60, in
dlopen_library("nvidia.cublas.lib", "libcublas.so*")
File "/opt/mamba/lib/python3.10/site-packages/deepmd/env.py", line 53, in dlopen_library
ctypes.CDLL(str(libs[0].absolute()))
File "/opt/mamba/lib/python3.10/ctypes/init.py", line 374, in init
self._handle = _dlopen(self._name, mode)
OSError: /opt/mamba/lib/python3.10/site-packages/nvidia/cublas/lib/libcublas.so.11: undefined symbol: cublasLt_for_cublas_HSS, version libcublasLt.so.11
However, when I import deepmd.infer using ipython, it works although gives some warnings, it can run. But my problem is that I need to run the py scripts, but I did not figure out what is the problem. Could you please help me? Thanks a lot !

[njzjz]

I tried to reproduce the issue in a new Colab environment. No errors occurred. See this link. The problem is more likely to be related to your environment.

[plumbum082]

Thank you very much! I have found the problem, it is indeed related to my environment, I have already have cuda toollkit installed in my environment and I fixed it with this command: pip uninstall nvidia_cublas_cu11