13、Ray Datasets:创建、保存与多工具集成使用指南

Ray Datasets:创建、保存与多工具集成使用指南

1. 创建和保存Ray数据集

可以通过调用 ray.data.from_items 从本地集合创建数据集,但本地集合会限制可处理的数据范围,Ray还支持其他多种方式。

1.1 Apache Arrow

Apache Arrow定义了一种独立于语言的列式内存格式,用于处理扁平数据和分层数据。其关键组件包括:
- 丰富的数据类型集,涵盖SQL和JSON类型,如int、BigInt、decimal、varchar、map、struct和array。
- 列式内存表示,支持基于定义的数据类型构建任意复杂的记录结构。
- 支持多种数据结构,如选择列表(类似枚举)、哈希表和队列。
- 使用共享内存、TCP/IP和远程直接内存访问(RDMA)进行进程间数据交换。
- 数据库,用于以多种语言(包括Java、C++、Python、Ruby、Rust、Go和JavaScript)读写列式数据。
- 各种操作的算法,如位图选择、哈希、过滤、分桶、排序和匹配。
- 通过列式内存压缩提高内存使用效率。
- 内存持久化工具,可通过非易失性内存、SSD或HDD实现短期持久化。

Ray使用Arrow将外部数据加载到数据集中,这些数据集支持多种文件格式和文件系统,目前支持的格式有CSV、JSON、Parquet、NumPy、文本和原始二进制。加载数据的函数遵循 read_[format] 模式,位于 ray.data 模块中,示例如下:


                
内容概要:本文详细比较了GPU、TPU专用AI芯片在大模型推理优化方面的性能、成本及适用场景。GPU以其强大的并行计算能力和高带宽显存,适用于多种类型的神经网络模型和计算任务,尤其适合快速原型开发和边缘计算设备。TPU专为机器学习设计,擅长处理大规模矩阵运算密集型任务,如Transformer模型的推理,具有高吞吐量和低延迟特性,适用于自然语言处理和大规模数据中心的推理任务。专用AI芯片通过高度定制化架构,针对特定神经网络模型进行优化,如卷积神经网络(CNN),在处理特定任务时表现出色,同时具备低功耗和高能效比的优势,适用于边缘计算设备。文章还介绍了各自的优化工具和框架,如CUDA、TensorRT、TPU编译器等,并从硬件成本、运营成本和开发成本三个角度进行了成本对比。 适合人群:从事人工智能、深度学习领域的研究人员和技术人员,尤其是对大模型推理优化感兴趣的读者。 使用场景及目标:①帮助读者理解GPU、TPU和专用AI芯片在大模型推理中的优缺点;②为选择适合的硬件平台提供参考依据,以实现最优的推理性能和成本效益;③介绍各种优化工具和框架,帮助开发者高效部署和优化模型。 其他说明:本文不仅涵盖了硬件架构特性,还深入探讨了优化技术和应用场景,旨在为读者提供全面的技术参考。在选择硬件平台时,需综合考虑具体任务需求、预算限制及开发资源等因素。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值