cunumpy

Install from PyPI with

pip install cunumpy

Simple wrapper for numpy and cupy. Replace import numpy as np with import cunumpy as xp.

Example usage:

export ARRAY_BACKEND=cupy

import cunumpy as xp
arr = xp.array([1,2])

print(type(arr))
print(xp.__version__)

Links: