首页
/ XGBoost 2.1.4版本发布:支持Blackwell架构GPU与CUDA 12.8+

XGBoost 2.1.4版本发布:支持Blackwell架构GPU与CUDA 12.8+

2025-05-06 22:40:14作者:苗圣禹Peter

XGBoost团队近期发布了2.1.4版本,这是一个重要的补丁更新,主要针对GPU支持进行了功能增强和兼容性改进。作为机器学习领域广泛使用的梯度提升框架,XGBoost持续优化其GPU加速能力,以满足高性能计算需求。

核心更新内容

本次2.1.4版本最值得关注的改进是对NVIDIA最新Blackwell架构GPU的支持。Blackwell是NVIDIA推出的新一代GPU架构,在AI和HPC工作负载方面有显著性能提升。XGBoost通过此更新确保了在新硬件上的兼容性和优化运行。

同时,该版本还完善了对CUDA 12.8及以上版本的支持。CUDA作为NVIDIA的并行计算平台,其新版本通常会带来性能优化和新特性。XGBoost保持与最新CUDA版本的兼容性,使开发者能够充分利用最新的GPU计算能力。

技术实现细节

在底层实现上,XGBoost团队通过多个PR完成了这些功能增强:

  1. GPU内存管理优化,确保在Blackwell架构上的高效内存使用
  2. CUDA内核函数更新,适配CUDA 12.8+的API变化
  3. 构建系统改进,支持新版工具链
  4. 测试套件扩展,覆盖新硬件和软件环境

这些改进不仅带来了新硬件的支持,还提升了框架在现有硬件上的稳定性和性能表现。

生态整合进展

XGBoost 2.1.4版本已全面覆盖主流分发渠道:

  • PyPI:Python包索引,供pip直接安装
  • Maven Central:Java生态系统的核心仓库
  • conda-forge:跨平台的conda包管理渠道

这种多平台支持确保了不同技术栈的用户都能方便地获取和使用新版本。特别是对于依赖GPU加速的研究和生产环境,及时更新到2.1.4版本可以获得更好的硬件兼容性和性能表现。

升级建议

对于需要使用Blackwell架构GPU或CUDA 12.8+的用户,建议尽快升级到XGBoost 2.1.4版本。升级过程与常规版本更新相同,可以通过各平台的包管理工具完成。在升级前,建议检查CUDA驱动版本和硬件兼容性,确保满足运行环境要求。

XGBoost团队持续致力于提升框架的性能和扩展性,这次更新再次体现了其对前沿硬件支持和技术生态整合的重视。

登录后查看全文
热门项目推荐
相关项目推荐

项目优选

收起
kernelkernel
deepin linux kernel
C
22
6
docsdocs
OpenHarmony documentation | OpenHarmony开发者文档
Dockerfile
161
2.05 K
nop-entropynop-entropy
Nop Platform 2.0是基于可逆计算理论实现的采用面向语言编程范式的新一代低代码开发平台,包含基于全新原理从零开始研发的GraphQL引擎、ORM引擎、工作流引擎、报表引擎、规则引擎、批处理引引擎等完整设计。nop-entropy是它的后端部分,采用java语言实现,可选择集成Spring框架或者Quarkus框架。中小企业可以免费商用
Java
8
0
openGauss-serveropenGauss-server
openGauss kernel ~ openGauss is an open source relational database management system
C++
146
191
leetcodeleetcode
🔥LeetCode solutions in any programming language | 多种编程语言实现 LeetCode、《剑指 Offer(第 2 版)》、《程序员面试金典(第 6 版)》题解
Java
60
16
ohos_react_nativeohos_react_native
React Native鸿蒙化仓库
C++
198
279
apintoapinto
基于golang开发的网关。具有各种插件,可以自行扩展,即插即用。此外,它可以快速帮助企业管理API服务,提高API服务的稳定性和安全性。
Go
22
0
RuoYi-Vue3RuoYi-Vue3
🎉 (RuoYi)官方仓库 基于SpringBoot,Spring Security,JWT,Vue3 & Vite、Element Plus 的前后端分离权限管理系统
Vue
949
556
ShopXO开源商城ShopXO开源商城
🔥🔥🔥ShopXO企业级免费开源商城系统,可视化DIY拖拽装修、包含PC、H5、多端小程序(微信+支付宝+百度+头条&抖音+QQ+快手)、APP、多仓库、多商户、多门店、IM客服、进销存,遵循MIT开源协议发布、基于ThinkPHP8框架研发
JavaScript
96
15
Cangjie-ExamplesCangjie-Examples
本仓将收集和展示高质量的仓颉示例代码,欢迎大家投稿,让全世界看到您的妙趣设计,也让更多人通过您的编码理解和喜爱仓颉语言。
Cangjie
346
1.33 K