CuPy — библиотека Python для GPU-ускоренных вычислений, совместимая с NumPy и SciPy. Она работает как замена NumPy/SciPy на платформах NVIDIA CUDA и AMD ROCm без изменения кода.
Библиотека предоставляет доступ к низкоуровневым функциям CUDA: RawKernels, Streams и прямые вызовы CUDA Runtime API для тонкой настройки производительности.
Готовые пакеты доступны через PyPI и Conda-Forge для Linux и Windows. Поддерживаются архитектуры x86_64, aarch64 и ppc64le. Разрабатывается компанией Preferred Networks и сообществом.
GitHub ★ 11,337
0 комментариев