NVIDIA英伟达嵌入式系统 · 2022年08月10日 · 北京市

DevZone | NVIDIA cuBLAS库

cuBLAS 库可提供基本线性代数子程序(BLAS)的 GPU 加速实现。cuBLAS 利用针对 NVIDIA GPU 高度优化的插入式行业标准 BLAS API,加速 AI 和 HPC 应用。cuBLAS 库包含用于批量运算、跨多个 GPU 的执行以及混合精度和低精度执行的扩展程序。通过使用 cuBLAS,应用将能自动从定期性能提升及新的 GPU 体系架构中受益。cuBLAS 库包含在 NVIDIA HPC SDK 和 CUDA 工具包中。

cuBLAS 多 GPU 扩展

cuBLASMg 提供了先进的多 GPU 矩阵间乘法,您可在多台设备间以 2D 块循环方式分发每个矩阵。cuBLASMg 目前已加入 CUDA 数学库抢先体验计划。

立即申请加入此体验计划:https://developer.nvidia.cn/c...

cuBLAS 性能

cuBLAS 库针对 NVIDIA GPU 上的性能进行了高度优化,并利用 Tensor Core 对低精度和混合精度矩阵乘法进行加速。

cuBLAS 的主要特性

  • 全面支持 152 个标准 BLAS 例程
  • 支持半精度和整数矩阵乘法
  • GEMM 和 GEMM 扩展程序针对 Volta 及 Turing Tensor Core 进行了优化
  • 针对各种深度学习模型中使用的规模调整 GEMM 性能
  • 支持用于并发操作的 CUDA 流

点击链接,了解更多 NVIDIA cuBLAS 库相关内容
https://developer.nvidia.cn/c...

NVIDIA 深度学习培训中心

我们的 NVIDIA 深度学习培训中心还推出了 NVIDIA cuBLAS 相关课程,您可以复制链接查看课程详情。

加速计算基础——CUDA C/C++

您将能够使用最基本的 CUDA 工具和技术,加速和优化仅适用于 CPU 的 C/C++ 应用程序。您将了解 CUDA 开发的迭代风格,这将帮助您快速发布加速应用程序。

详情请见:https://courses.nvidia.com/co...

加速计算基础——CUDA Python

您将了解使用 CUDA 和 Numba 编译器在大规模并行 GPU 上加速运行 Python 应用程序的基本工具和技能。

详情请见:https://courses.nvidia.com/co...

通过 CUDA C++ 在多个 GPU 之间扩展工作负载

您将学习如何在单个节点上使用多个 GPU,构建强大高效的 CUDA C++ 应用程序。

详情请见:https://courses.nvidia.com/co...

通过并发流加速 CUDA C++ 应用程序

您将在 CUDA C++ 应用程序中,学习利用 CUDA Streams 进行复制/计算重叠。

详情请见:https://courses.nvidia.com/co...

推荐阅读
目录
极术微信服务号
关注极术微信号
实时接收点赞提醒和评论通知
安谋科技学堂公众号
关注安谋科技学堂
实时获取安谋科技及 Arm 教学资源
安谋科技招聘公众号
关注安谋科技招聘
实时获取安谋科技中国职位信息