Ampere, 80 ГБ HBM2e

NVIDIA A100 80GB

Архитектура Ampere и 80 ГБ HBM2e — вдвое больше памяти, чем у A100 40GB, и пропускная способность до 2 ТБ/с. Для моделей, которые не помещаются в 40 ГБ.

Характеристики

Спецификации

Архитектура
NVIDIA Ampere
VRAM
80 GB HBM2e
Пропускная способность
1.9–2.0 TB/s
FP16 Tensor Core (dense)
312 TFLOPS
Интерконнект
NVLink 600 GB/s
Потребление (TDP)
300–400W
Сравнение

Почему именно 80GB?

A100 40GB и 80GB различаются не только объёмом, но и типом памяти: у 80-гигабайтной версии HBM2e и 1.9–2.0 ТБ/с против HBM2 и 1.55 ТБ/с. Вычислительная часть одинаковая. Диапазоны в характеристиках — от PCIe- до SXM-версии.

  • Llama 3 70B в 8-битной квантизации: веса занимают около 70 ГБ, на KV-кэш остаётся порядка 10 ГБ
  • Вдвое больше памяти под батч и активации при обучении
  • NVLink 600 ГБ/с между картами: в SXM-серверах и через мост на две PCIe-карты

Сравнение версий

ХарактеристикаA100 40GBA100 80GB
Тип памятиHBM2HBM2e
Пропускная способность1.55 TB/s1.9–2.0 TB/s
Предел по весам в FP16~20B params~40B params
Предел по весам: 2 байта на параметр в FP16. KV-кэш и активации занимают дополнительную память, поэтому на практике модель должна быть меньше.
Решения

Для каких задач

Дообучение LLM

LoRA для моделей на 7–13B параметров в BF16 с запасом под батч. QLoRA — вплоть до 70B: в 4 битах её веса занимают около 35 ГБ.

Батч-инференс

80 ГБ оставляют место под большой батч и KV-кэш, а до 2 ТБ/с памяти ускоряют генерацию токенов, которая упирается в чтение весов.

Data Science

Обработка табличных данных на GPU (RAPIDS) и графовые нейросети, которым нужен большой объём памяти.

Нужны 80 ГБ HBM2e?

Арендуйте A100 80GB поминутно: драйверы и CUDA уже установлены, сервер поднимается за несколько минут.