CuPy: NumPy & SciPy for GPU - A NumPy/SciPy-compatible array library for GPU-accelerated computing with Python, specifically built for AMD ROCm 4.3 platform
—
Loading evals
Install with Tessl CLI
npx tessl i tessl/pypi-cupy-rocm-4-3