NVIDIA CUDA-X 库
NVIDIA CUDA-Xâ„¢ 库基于 CUDA® 构建，是一系列库的集åˆï¼Œä¸Žä»…使用 CPU 的替代方案相比，å¯åœ¨åŒ…括 AI 和高性能计算在内的应用领域ä¸æ供显著更高的性能。
从资æºå—é™çš„物è”网设备到自动驾驶汽车，å†åˆ°å…¨çƒæœ€å¤§çš„超级计算机，NVIDIA åº“æ— å¤„ä¸åœ¨ã€‚å› æ¤ï¼Œç”¨æˆ·å¯ä»¥é€šè¿‡ä¸æ–æ‰©å±•çš„ç®—æ³•é›†èŽ·å¾—é«˜åº¦ä¼˜åŒ–çš„å®žçŽ°ã€‚æ— è®ºæ˜¯æž„å»ºæ–°åº”ç”¨è¿˜æ˜¯åŠ é€ŸçŽ°æœ‰åº”ç”¨ï¼Œå¼€å‘者都å¯ä»¥åˆ©ç”¨ NVIDIA 库，以最简å•çš„æ–¹å¼å¼€å§‹ GPU åŠ é€Ÿã€‚
组件
CUDA æ•°å¦åº“
GPU åŠ é€Ÿçš„æ•°å¦åº“为分å动力å¦ã€è®¡ç®—æµä½“动力å¦ã€è®¡ç®—化å¦ã€åŒ»å¦æˆåƒå’Œåœ°éœ‡å‹˜æŽ¢ç‰é¢†åŸŸçš„è®¡ç®—å¯†é›†åž‹åº”ç”¨å¥ å®šäº†åŸºç¡€ã€‚
Python ä¸çš„ NVIDIA æ•°å¦åº“
为 Python 生æ€ç³»ç»Ÿå¯ç”¨ GPU åŠ é€Ÿçš„æ•°å¦è¿ç®—。
nvmath-python
nvmath-python (测试版) 是一个开æºåº“，å¯å¯¹ NVIDIA æ•°å¦åº“ä¸çš„æ ¸å¿ƒæ•°å¦è¿ç®—进行高性能访问。
了解详情
科å¦è®¡ç®—库
适用于需è¦éµå®ˆæ•°å¦å¯¹ç§°æ€§çš„神ç»ç½‘络的应用，特别是分å结构ã€è›‹ç™½è´¨å’Œææ–™ç‰ 3D å‡ ä½•æ•°æ®çš„ç‰æ–¹å·®ã€‚
cuEquivariance
å¼€æº Python åº“ï¼Œæ—¨åœ¨åŠ é€Ÿç‰å˜ç¥žç»ç½‘ç»œçš„æž„å»ºå’Œæ‰§è¡Œï¼Œå°¤å…¶æ˜¯å¤„ç† 3D 空间ä¸æ•°æ®æ—‹è½¬å’Œå¹³ç§»çš„神ç»ç½‘络。
了解详情并行算法库
GPU åŠ é€Ÿçš„é«˜æ•ˆå¹¶è¡Œç®—æ³•åº“ï¼Œç”¨äºŽ C++ ä¸çš„多项è¿ç®—，以åŠåœ¨ç ”究自然科å¦ã€ç‰©æµã€æ—…行规划ç‰é¢†åŸŸçš„关系时与图形一起使用。
Thrust
GPU åŠ é€Ÿçš„ C++ 并行算法和数æ®ç»“构库。
了解详情计算光刻库
旨在应对纳米级计算光刻的现代挑战。
cuLitho
包å«ä¼˜åŒ–工具和算法的库，å¯ä½¿ç”¨ GPU åŠ é€Ÿè®¡ç®—å…‰åˆ»å’ŒåŠå¯¼ä½“åˆ¶é€ ã€‚
了解详情é‡å库
为é‡å计算实现仿真ã€HPC 集æˆå’Œ AI。
cuQuantum
NVIDIA cuQuantum æ˜¯ä¸€ç»„ç”¨äºŽåŠ é€Ÿé‡å计算模拟的高度优化库。
cuPQC
ç”¨äºŽåŠ é€ŸåŽé‡ååŠ å¯† (PQC) 工作æµç¨‹çš„优化库 SDK。
æ•°æ®å¤„ç†åº“
GPU åŠ é€Ÿåº“ï¼Œå¯åŠ é€Ÿè¡¨æ ¼ã€æ–‡æœ¬å’Œå›¾åƒæ•°æ®çš„æ•°æ®å¤„ç†å·¥ä½œæµç¨‹ã€‚
RAPIDS cuDF
æ— éœ€æ›´æ”¹ä»£ç å³å¯åŠ é€Ÿè¡¨æ ¼æ•°æ®ï¼ŒåŒ…括 pandas å’Œ Polars。
RAPIDS cuML
æ— éœ€æ›´æ”¹ä»£ç ，å³å¯åœ¨ scikit-learnã€UMAP å’Œ HDBSCAN ä¸åŠ 速 ML 算法。
RAPIDS cuGraph
借助 GPU åŠ é€Ÿçš„ NetworkX æ‰©å±•å’ŒåŠ é€Ÿå›¾å½¢åˆ†æžã€‚
NVIDIA cuVS
应用 cuVS 算法æ¥åŠ 速数æ®æŒ–掘和è¯ä¹‰æœç´¢åº”用的å‘é‡æœç´¢ï¼ŒåŒ…括 GPU 原生近邻算法 CAGRA 的出色性能。
NeMo Curator
NVIDIA NeMo Curator 通过大规模处ç†æ–‡æœ¬ã€å›¾åƒå’Œè§†é¢‘æ•°æ®è¿›è¡Œè®ç»ƒå’Œè‡ªå®šä¹‰ï¼Œæ高生æˆå¼ AI 模型的准确性。它还æ供用于生æˆåˆæˆæ•°æ®çš„预构建æµç¨‹ï¼Œä»¥å®šåˆ¶å’Œè¯„估生æˆå¼ AI 系统。
Morpheus
开放å¼åº”用框架，å¯ä¼˜åŒ–网络安全 AI 工作æµï¼Œä»¥åˆ†æžå¤§é‡å®žæ—¶æ•°æ®ã€‚
GPU Direct Storage
NVIDIA GPUDirect® Storage å¯åœ¨æœ¬åœ°æˆ–远程å˜å‚¨ (例如 NVMe 或 NVMe over Fabric (NVMe-oF)) 与 GPU 显å˜ä¹‹é—´åˆ›å»ºç›´æŽ¥æ•°æ®è·¯å¾„。
Dask
使用 Dask 上的 RAPIDS 将数æ®ç§‘å¦å·¥ä½œæµæ‰©å±•åˆ°å¤šä¸ªèŠ‚点。
适用于 Apache Spark çš„ RAPIDS åŠ é€Ÿå™¨
åªéœ€æ›´æ”¹æœ€å°‘的代ç ，å³å¯åŠ 速现有的 Apache Spark 应用。
图åƒå’Œè§†é¢‘库
GPU åŠ é€Ÿåº“ï¼Œç”¨äºŽä½¿ç”¨ CUDA å’Œ GPU 的专用硬件组件进行图åƒå’Œè§†é¢‘解ç ã€ç¼–ç 和处ç†ã€‚
RAPIDS cuCIM
åŠ é€Ÿ n ç»´ (尤其是生物医å¦å›¾åƒ) 的输入/ 输出 (IO) ã€è®¡ç®—机视觉和图åƒå¤„ç†ã€‚
CV-CUDA
å¼€æºåº“，用于视觉 AI 工作æµä¸çš„高性能ã€GPU åŠ é€Ÿçš„é¢„å¤„ç†å’ŒåŽå¤„ç†ã€‚
NVIDIA DALI
å¯ç§»æ¤çš„å¼€æºåº“，用于解ç 和增强图åƒå’Œè§†é¢‘ï¼Œä»¥åŠ é€Ÿæ·±åº¦å¦ä¹ 应用程åºã€‚
nvJPEG
用于 JPEG 解ç 的高性能 GPU åŠ é€Ÿåº“ã€‚
NVIDIA 性能基元
GPU åŠ é€Ÿçš„å›¾åƒã€è§†é¢‘和信å·å¤„ç†åŠŸèƒ½ã€‚
NVIDIA 视频编解ç 器 SDK
在 Windows å’Œ Linux ä¸Šæ‰§è¡Œç¡¬ä»¶åŠ é€Ÿçš„è§†é¢‘ç¼–ç 和解ç 。
NVIDIA å…‰æµ SDK
展示 NVIDIA GPU 的最新硬件功能，专用于计算图åƒä¹‹é—´åƒç´ 的相对è¿åŠ¨ã€‚
通信库
性能ç»è¿‡ä¼˜åŒ–的多 GPU 和多节点通信基元。
NVSHMEM
适用于 GPU 显å˜çš„ OpenSHMEM æ ‡å‡†ï¼Œä»¥åŠç”¨äºŽæ高 GPU 性能的扩展程åºã€‚
了解详情
NCCL
用于快速多 GPUã€å¤šèŠ‚点通信的开æºåº“，å¯åœ¨ä¿æŒä½Žå»¶è¿Ÿçš„åŒæ—¶æ›´å¤§é™åº¦åœ°æ高带宽。
了解详情
深度å¦ä¹ æ ¸å¿ƒåº“
适用于使用 CUDA å’Œ GPU 专用硬件组件的深度å¦ä¹ 应用程åºçš„ GPU åŠ é€Ÿåº“ã€‚
NVIDIA CUTLASS
针对 CUDA 执行和内å˜å±‚次结构ä¸çš„é«˜æ€§èƒ½çº¿æ€§ä»£æ•°å’Œå¼ é‡è¿ç®—优化抽象。
NVIDIA TensorRT-LLM
用于生产部署的高性能深度å¦ä¹ 推ç†ä¼˜åŒ–器和è¿è¡Œæ—¶ã€‚
NVIDIA cuDNN
GPU åŠ é€Ÿçš„æ·±åº¦ç¥žç»ç½‘络基元库。
åˆä½œä¼™ä¼´åº“
OpenCV
GPU åŠ é€Ÿçš„å¼€æºåº“，用于计算机视觉ã€å›¾åƒå¤„ç†å’Œæœºå™¨å¦ä¹ ，现在支æŒå®žæ—¶æ“作。
FFmpeg
包å«éŸ³é¢‘和视频处ç†æ’件库的开æºå¤šåª’体框架。
ArrayFire
GPU åŠ é€Ÿçš„å¼€æºåº“，用于矩阵ã€ä¿¡å·å’Œå›¾åƒå¤„ç†ã€‚
MAGMA
ç”± Magma å¼€å‘的用于异构架构的 GPU åŠ é€Ÿçº¿æ€§ä»£æ•°ä¾‹ç¨‹ã€‚
IMSL Fortran 数值库
ç”± RogueWave æ供的 GPU åŠ é€Ÿå¼€æº Fortran 库，包å«æ•°å¦ã€ä¿¡å·å’Œå›¾åƒå¤„ç†ä»¥åŠç»Ÿè®¡åŠŸèƒ½ã€‚
Gunrock
专为 GPU 设计的图形处ç†åº“。
CHOLMOD
GPU åŠ é€Ÿçš„ç¨€ç–直接求解器函数，包å«åœ¨ SuiteSparse 线性代数包ä¸ã€‚
Triton Ocean SDK
Triton æ供的海洋ã€æ¸¸æˆä¸çš„水体ã€ä»¿çœŸå’Œè®ç»ƒåº”用的实时视觉仿真。
CUVIlib
ç”¨äºŽåŠ é€ŸåŒ»ç–—ã€å·¥ä¸šå’Œå›½é˜²é¢†åŸŸæˆåƒåº”用的基元。
资æº
文档
培è®
社区
开始使用
NVIDIA å¼€å‘者计划æˆå‘˜å¯ä»¥æŠ¢å…ˆä½“验所有 CUDA åº“ç‰ˆæœ¬ä»¥åŠ NVIDIA 在线错误报告和功能请求系统。