ДаліНаступний посібник
Основний ML для розгортання Apple на пристрої
технічний
Технічний КЕРІВНИЦТВО
Тензорні ядра — це спеціалізовані апаратні блоки в сучасних графічних процесорах NVIDIA, які надзвичайно швидко виконують операції множення та накопичення матриць.
Послухайте цей посібникШІ-розказаний · ~2 min
They are the main reason a single GPU can train and run large neural networks orders of magnitude faster than general-purpose compute would allow.
Представлені в архітектурі Volta у 2017 році, Tensor Cores — це спеціальні схеми, які обчислюють множення невеликої матриці плюс додавання (D = A x B + C) за одну операцію, замість того, щоб виконувати кожне множення по одному на стандартних ядрах CUDA. Оскільки практично кожен рівень нейронної мережі зводиться до множення матриці, це відповідає математичним показникам, які насправді потребує ШІ. Кожне покоління GPU розширювало те, що вони обробляли: Volta створила плитки 4x4 FP16, тоді як пізніші архітектури Ampere, Hopper і Blackwell додали формати нижчої точності, такі як TF32, BF16, INT8, FP8 і FP4. Нижча точність означає більше чисел, що обробляються за такт, значно підвищуючи пропускну здатність для навчання та висновків, зберігаючи при цьому прийнятну точність.
Навчання великих мовних моделей, таких як трансформатори у стилі GPT, де мільярди множень матриць за крок виконуються на тензорних ядрах у BF16 або FP8.
Запуск висновків у реальному часі для чат-ботів і генераторів зображень із використанням квантування INT8 або FP8 для обслуговування більшої кількості користувачів на GPU.
Прискорення NVIDIA DLSS у відеоіграх, де нейронна мережа покращує кадри з нижчою роздільною здатністю за допомогою тензорних ядер кожного кадру.
Прискорення наукових обчислень, таких як згортання білків (AlphaFold) і моделі погоди, які були переформульовані як нейронні навантаження з великими матрицями.
Тензорні ядра продовжують рухатися до дедалі нижчої точності: Хоппер додав FP8, а Блекуелл представив 4-розрядний FP4 із масштабуванням, керованим апаратним забезпеченням, приблизно подвоюючи пропускну здатність на кожному кроці для важких робочих навантажень. Очікуйте більш жорсткої підтримки розрідженості (пропуск нульових ваг), форматів мікромасштабування, які додають масштабні коефіцієнти до невеликих блоків чисел, і більш глибокої інтеграції з системами пам’яті, щоб ядра залишалися живими. У міру зростання моделей основним полем битви за продуктивність апаратного забезпечення штучного інтелекту залишається матричний механізм, а не сира тактова частота.
Безплатний розсилок
Три перевірені історії про ШІ щоробочого ранку, написані простою мовою. Безплатно назавжди, без реклами.
Один лист щотижня з понеділка по п'ятницю. Відписатися в один клік. Ми ніколи не продаємо та не передаємо вашу адресу.
Перевір себе
Миттєва зворотна связь на кожну відповідь та сертифікат із перевіреним ідентифікатором, яким можна поділитися, після проходження курсу.
Підтримайте безплатну освіту у сфері ШІ. AI Understanding — це некомерційна організація 501(c)(3) — без реклами, без платних стін, назавжди. Зробіть пожертву
Тензорні ядра — це спеціалізовані апаратні блоки в сучасних графічних процесорах NVIDIA, які надзвичайно швидко виконують операції множення та накопичення матриць. Вони є основною причиною того, що один графічний процесор може навчати та запускати великі нейронні мережі на порядки швидше, ніж це дозволяють обчислення загального призначення.
Тензорні ядра виконують множення злитої матриці плюс накопичення за один крок, що є саме тією операцією, яка домінує на рівнях нейронної мережі.
Tensor Cores дебютувала з архітектурою Volta у 2017 році, починаючи з операцій матриці FP16 4x4.
Використання меншої кількості бітів на число означає, що через апаратне забезпечення кожного циклу протікає більше значень, значно збільшуючи швидкість лише з невеликою, зазвичай допустимою, втратою точності.
Вхідні дані можуть мати низьку точність, але накопичувач зазвичай працює з вищою точністю, наприклад FP32, щоб обмежити накопичення помилок округлення.
Базові бібліотеки автоматично розбивають великі матриці на фрагменти розміром із Tensor-Core, тому фреймворки отримують прискорення без кодування вручну.
Продовжуйте вчитися
Інші посібники, вибрані для цієї теми
ДаліНаступний посібник
Основний ML для розгортання Apple на пристрої
технічний