import functools as _functools
import sys as _sys
import warnings as _warnings
import numpy as _numpy
from cupy import _environment
from cupy import _version
_environment._detect_duplicate_installation() # NOQA
_environment._setup_win32_dll_directory() # NOQA
_environment._preload_library('cutensor') # NOQA
_environment._preload_library('nccl') # NOQA
try:
from cupy import _core # NOQA
except ImportError as e:
raise ImportError(f'''
================================================================
{_environment._diagnose_import_error()}
Original error:
{type(e).__name__}: {e}
================================================================
''') from e
from cupy import cuda # NOQA
# Do not make `cupy.cupyx` available because it is confusing.
import cupyx as _cupyx # NOQA
"""
cuDNN Wrapper
Use `cupy_backends.cuda.libs.cudnn` directly in CuPy codebase.
"""
from cupy import _environment
available = True
try:
_environment._preload_library('cudnn')
from cupy_backends.cuda.libs.cudnn import * # NOQA
except ImportError as e:
available = False
_environment._preload_warning('cudnn', e)