Registry / ai-ml / cutensor-cu13

cutensor-cu13

JSON →
library2.6.0pypypi✓ verified 81d ago

Python bindings for NVIDIA cuTENSOR, a GPU-accelerated tensor computation library optimized for deep learning and HPC. Version 2.6.0 supports CUDA 13.x. Release cadence follows cuTENSOR releases (major.minor.patch).

pip install cutensor-cu13
INSTALL
IMPORT
SIG · CUTENSOR-CU13
C
cutensor-cu13
ai-mlpythonv2.6.0
harness data pending
Install & Compatibility
Where this runs

No compatibility data collected yet for this library.

Code
Verified usage

Verified import paths — ran on the pinned version, not inferred.

tensor
import cutensor
from cutensor import tensor

Basic tensor creation and matrix multiplication on GPU.

import numpy as np from cutensor import tensor # Create two tensors on GPU (requires CUDA context) a = tensor(np.random.rand(3,4), device='cuda') b = tensor(np.random.rand(4,5), device='cuda') # Perform matrix multiplication c = a @ b print(c)
Debug
Known issues
breakingThe Python API changed in cuTENSOR 2.0+; the old 'cutensor.init' pattern was removed.
fix
Use 'from cutensor import tensor' directly; no explicit init call needed.
affects: <2.0.0
gotchaInput tensors must be contiguous in memory; non-contiguous arrays may cause silent wrong results or crashes.
fix
Call np.ascontiguousarray() on NumPy arrays before converting to cutensor tensors.
affects: All
deprecatedUsing 'cutensor.TensorHandle' is deprecated in favor of 'cutensor.tensor'.
fix
Replace 'TensorHandle' with 'tensor' class.
affects: >=2.0.0
Upgrade
Version history
2.6.0latest on PyPI · released Mar 17, 2026
Audit
Dependencies
nvidia-cuda-runtime-cu13requiredRequired CUDA runtime libraries for GPU operations.
nvidia-cublas-cu13requiredRequired for BLAS operations used internally by cuTENSOR.
Agent activity
15 hits · last 30 days
node
14
OpenAI (training)
1
Resources
cutensor-cu13 — pip install cutensor-cu13 · libregistry