fix cutensor import
@@ -6,14 +6,14 @@ import warnings
import cupy
from cupy.core import _accelerator
from cupy import _util
-from cupy.cuda import cutensor as cuda_cutensor
from cupy.linalg._einsum_opt import _greedy_path
from cupy.linalg._einsum_opt import _optimal_path
-if cuda_cutensor.available:
+try:
+ import cupy_backends.cuda.libs.cutensor # NOQA
from cupy import cutensor
-else:
+except ImportError:
cutensor = None
