'''
python -m pip install -U "cupy-cuda13x[ctk]"
pip install cupy-cuda13x
'''


import cupy as cp

x = cp.random.random(1024*1024)
y = cp.fft.fft(x)

print(y[:5])
print("GPU:", cp.cuda.runtime.getDeviceProperties(0)["name"])
print("OK")