3 ms·
What was the difference in runtime performance, and did you try CuPy? https://github.com/cupy/cupy https://github.com/cupy/cupy : > CuPy is a NumPy/SciPy-comp
by westurner 3y ago
What was the difference in runtime performance, and did you try CuPy?
https://github.com/cupy/cupy https://github.com/cupy/cupy :
> CuPy is a NumPy/SciPy-compatible array library for GPU-accelerated computing with Python. CuPy acts as a drop-in replacement to run existing NumPy/SciPy code on NVIDIA CUDA or AMD ROCm platforms.
Projects using CuPy:
https://github.com/cupy/cupy/wiki/Projects-using-CuPy https://github.com/cupy/cupy/wiki/Projects-using-CuPy