diff --git a/python/cudaq/__init__.py b/python/cudaq/__init__.py index 376b20c8508..16fbd448fbc 100644 --- a/python/cudaq/__init__.py +++ b/python/cudaq/__init__.py @@ -24,7 +24,11 @@ cutensornet_libs = get_library_path(f"cutensornet-cu{cuda_major}") cutensornet_path = os.path.join(cutensornet_libs, "libcutensornet.so.2") - os.environ["CUDAQ_DYNLIBS"] = f"{custatevec_path}:{cutensornet_path}" + cutensor_libs = get_library_path(f"cutensor-cu{cuda_major}") + cutensor_path = os.path.join(cutensor_libs, "libcutensor.so.2") + + os.environ[ + "CUDAQ_DYNLIBS"] = f"{custatevec_path}:{cutensornet_path}:{cutensor_path}" # The following package is only available on `x86_64` (not `aarch64`). For # `aarch64`, the library must be provided another way (likely with