NVIDIA 发布 CUDA Python 1.0:Python 成 CUDA 一等公民
PyTorch · x · 2026-09-09
随 CUDA 13.3 发布,NVIDIA 推出 CUDA Python 1.0,将 Python 确立为 CUDA 平台受支持的一等使用方式,各组件独立版本化并提供语义化版本保证。
要点:
- cuda.core 1.0.0 提供设备、流、缓冲区、green contexts、进程检查点与进程间共享等 CUDA 运行时能力的 Pythonic 接口,异常式错误处理
- cuda.compute 1.0.0 暴露 CCCL 并行算法(sort/scan/reduce),可针对当前 GPU 编译并接受 Python lambda 自定义
- cuda.bindings 13.3.0 对 CUDA C API 提供 1:1 完整覆盖,供库作者使用;Numba 及新的 Numba CUDA MLIR 后端支持用 Python 写 SIMT kernel
- PyTorch、CuPy 及 NVIDIA 自家通信与数学库均已基于共享的 cuda.core 基础构建,减少版本冲突与互操作 bug
- 安装:pip install cuda-python cuda-cccl numba-cuda-mlir[cu13],nvmath-python 单独安装
「Infra」频道最新
- DeepSpeed 发布 AutoEP/AutoTP 等并行化进展并办线上答疑 — PyTorch · 2026-10-27
- Cloudflare Durable Objects 被验证为 AI agent 高并发场景优选架构 — tristanbob · 2026-10-27
- Gary Marcus:补上 Q3 数据后,Anthropic 算力效率增速现拐点 — GaryMarcus · 2026-10-27
- 推理社区与 K8s 社区高度重合:万亿市场的 Kubernetes 翻版 — matt_slotnick · 2026-10-27
- vLLM 考虑将更多栈迁往 Rust,推理引擎 Rust 化成趋势 — AccBalanced · 2026-10-27
- 同一套 DeepSeek 内核两种优化路线:vLLM 拆专家、SGLang 降访存 — AccBalanced · 2026-10-27