Как стать автором
Обновить

Вычисления с GPU-ускорением на Python

Уровень сложностиСложный
Время на прочтение9 мин
Количество просмотров10K
Всего голосов 33: ↑28 и ↓5+37
Комментарии4

Комментарии 4

На google colab предлагается tpu v2-8 с огромной вычислительной мощностью в узких задачах (до 25 Тфлопс?). Есть ли библиотеки под него?

Это перевод, отвечать некому.
А начало статьи подсказывает что автору помогал ИИ. :)

<тема статьи> – это важнейший компонент компьютера, отвечающий за отображение картинок и видео.  

tensorflow и pytorch поддерживают гугловские tpu.

до 25 Тфлопс?

Официально на одно ядро до 22.5 TFLOPS, а их всего 8 у v2-8 - то есть в сумме 180 TFLOPS, но по факту получается меньше, но все равно больше чем у T4, но есть нюансы.

Всем интересующимся запуском Python-кода на GPU рекомендую также обратить внимание на модуль Numba, который, помимо компиляции функций в нативный код, имеет богатый и очень простой в использовании инструментарий для работы с CUDA

Зарегистрируйтесь на Хабре, чтобы оставить комментарий