Есть много способов сделать это в зависимости от того, какие вычисления вы хотите сделать, и насколько глубоко вы готовы погрузиться в интуицию GPU.
Если вы используете графический процессор NVIDIA с CUDA, вы можете использовать библиотеку NVBLAS , поставляемую с инструментарием CUDA, в качестве сменной замены, ускоряющей функции BLAS уровня 3.
В дистрибутиве Anaconda Python есть пакет под названием Accelerate , который предлагает набор функций для ускорения GPU с использованием графических процессоров NVIDIA.
Пакет pycuda является еще одним вариантом.