CuPy — это библиотека для работы с массивами в стиле NumPy, но с поддержкой вычислений на GPU.

Она позволяет использовать NVIDIA CUDA для ускорения операций с массивами, что делает её полезной для научных вычислений, машинного обучения и других задач, требующих высокой производительности. 🐈

🟢Полная совместимость с NumPy (можно легко заменить numpy на cupy в коде).
🟢Аппаратное ускорение на GPU через CUDA.
🟢Поддержка многопоточных вычислений.
🟢Поддержка FFT, линейной алгебры, разреженных матриц и других численных методов.

Если у тебя есть видеокарта NVIDIA и установленная CUDA, ты можешь использовать CuPy для ускорения кода, который обычно работает на CPU через NumPy.

🔜 https://github.com/cupy/cupy

👉 @PythonPortal | #ресурсы