Usage

Using cuPyNumeric as a replacement for NumPy is simple. Replace your NumPy import statement with cuPyNumeric:

import numpy as np

becomes

import cupynumeric as np

Then, run the application like you usually do. For example, if you had a script main.py written in NumPy that adds two vectors,

import numpy as np
x = np.array([1.0, 2.0, 3.0, 4.0])
y = np.array([4.0, 3.0, 2.0, 1.0])
z = x + y
print(z)

change the import statement to use cuPyNumeric like below,

import cupynumeric as np
x = np.array([1.0, 2.0, 3.0, 4.0])
y = np.array([4.0, 3.0, 2.0, 1.0])
z = x + y
print(z)

And run the program, like this

python main.py

By default this invocation will use all the hardware resources (e.g. CPU cores, RAM, GPUs) available on the current machine.

For more information on controlling the resource allocation, running on multiple nodes etc. see https://docs.nvidia.com/legate/latest/usage.html.