cuQuantum installation

https://docs.nvidia.com/cuda/cuquantum

NVIDIA cuQuantum SDK is a high-performance library for quantum information science and beyond. Currently its primary target is quantum circuit simulations and it consists of two major components:

• cuStateVec: a high-performance library for state vector computations
• cuTensorNet: a high-performance library for tensor network computations

In addition to C APIs, cuQuantum also provides Python APIs via cuQuantum Python.

NVIDIA cuQuantum Appliance offers a containerized solution, including a multi-GPU backend for Google’s qsim state vector simulator.

Download: https://developer.nvidia.com/cuQuantum-downloads

Conda installation:

conda install -c conda-forge cuquantum-python
conda install -c conda-forge/label/broken cuquantum-python

Hello world:

from cuquantum import contract
from numpy.random import rand

a = rand(96,64,64,96)
b = rand(96,64,64)
c = rand(64,96,64)

r = contract("mhkn,ukh,xuy->mxny", a, b, c)
print(r.shape)
# (96, 64, 96, 64)

OKAY