Описание тега cupy

Описание тега Вопросы с тегом
CuPy - это реализация многомерного массива, совместимого с NumPy, на CUDA.

О CuPy

С домашней страницы CuPy:

Высокая производительность с CUDA

CuPy - это матричная библиотека с открытым исходным кодом, ускоренная с помощью NVIDIA CUDA. Он также использует библиотеки, связанные с CUDA, включая cuBLAS, cuDNN, cuRand, cuSolver, cuSPARSE, cuFFT и NCCL, чтобы полностью использовать архитектуру графического процессора.

Высокая совместимость с NumPy

Интерфейс CuPy полностью совместим с numpy; в большинстве случаев его можно использовать в качестве замены. Все, что вам нужно сделать, это просто заменитьnumpy с cupyв вашем коде Python. Он поддерживает различные методы, индексацию, типы данных, трансляцию и многое другое.

CuPy состоит из основного класса многомерных массивов, cupy.ndarray, и многие функции на нем. Он поддерживает подмножествоnumpy.ndarray интерфейс.

Ресурсы