Oracle自治数据库上的机器学习笔记本现已支持GPU
引言
近日,Oracle宣布其自治数据库Serverless版本现已为Oracle机器学习(OML)笔记本提供集成的OCI GPU(图形处理单元)实例访问支持。这一重大更新将为Python用户带来前所未有的便利,让他们能够轻松地在单个笔记本级别激活GPU资源,并利用支持GPU的Python包。
主要特性
-
GPU支持: 用户现在可以在Oracle自治数据库Serverless(搭载Oracle Database 23ai)上使用GPU进行计算密集型任务。
-
简化的GPU激活: 通过选择笔记本服务级别为"gpu",用户可以一键激活GPU资源。
-
自定义conda环境: 用户可以创建自定义conda环境,访问特定项目所需的GPU启用包。
-
多样化应用场景: 支持AI向量搜索的批量向量嵌入生成、深度学习模型构建以及批量推理等用例。
GPU的重要性
GPU在并行处理计算密集型任务方面表现出色,尤其适用于AI嵌入模型、深度学习建模和推理。它们可以显著提升各种用例的性能,同时在硬件成本方面保持相对经济实惠。
在自治数据库上启用GPU的步骤
- ADMIN用户创建自定义conda环境。
- 创建或打开笔记本,将笔记本类型(服务级别)更改为"gpu"。
- 下载并激活conda环境,然后运行Python代码。
- 使用GPU启用包的代码将自动受益于GPU加速。
需要注意的是,使用GPU需要付费账户,且Oracle机器学习应用的ECPU分配设置为16或更高。
未来规划
Oracle还公布了自治数据库GPU支持的路线图,包括:
- OML4Py嵌入式Python执行支持
- OML4Py空间AI增强
- ONNX格式模型的GPU卸载
- 多种GPU芯片选项支持
结语
这一更新无疑将大大提升Oracle自治数据库用户的机器学习和AI开发体验。如果您正在使用Oracle自治数据库(ADW, ATP)实例且搭载Oracle Database 23ai,现在就可以尝试创建带有GPU支持包的conda环境,并启用笔记本的GPU资源。
欢迎大家在评论区分享您对这一新功能的看法和使用体验!
参考资源
注:本文基于Oracle官方博客发布的信息编写,最终功能以Oracle官方发布为准。