CuPy — это открытая замена NumPy, которая переносит вычисления на GPU. Во многих случаях достаточно заменить всего одну строку:
import numpy as np
на
import cupy as cp
И тот же код может начать выполняться на видеокарте, ускоряя вычисления в десятки раз. В некоторых задачах разработчики заявляют прирост производительности до 100×.
Что умеет:
- совместим с большинством кода на NumPy и SciPy;
- не требует переписывать программу;
- поддерживает NVIDIA CUDA и AMD ROCm;
- полностью открытый исходный код.
Если работаете с анализом данных, машинным обучением или тяжёлыми вычислениями на Python — проект определённо стоит добавить в закладки.
GitHub: https://github.com/cupy/cupy #программирование #полезное