cuML 正在参加 2020 年度 OSC 中国开源项目评选,请投票支持!
cuML 在 2020 年度 OSC 中国开源项目评选 中已获得 {{ projectVoteCount }} 票,请投票支持!
投票让它出道
已投票
授权协议: Apache
开发语言: C/C++
操作系统: 跨平台
收录时间: 2019-08-06
提 交 者: 红薯

cuML是一套用于实现与其他RAPIDS项目共享兼容API的机器学习算法和数学原语函数。

cuML使数据科学家、研究人员和软件工程师能够在GPU上运行传统的表格ML任务,而无需深入了解CUDA编程的细节。 在大多数情况下,cuML的Python API与来自scikit-learn的API相匹配。

对于大型数据集,这些基于GPU的实现可以比其CPU等效完成10-50倍。 有关性能的详细信息,请参阅cuML基准测试笔记本。

例如,以下Python代码段加载输入并计算DBSCAN集群,所有这些都在GPU上:

import cudf
from cuml.cluster import DBSCAN

# Create and populate a GPU DataFrame
gdf_float = cudf.DataFrame()
gdf_float['0'] = [1.0, 2.0, 5.0]
gdf_float['1'] = [4.0, 2.0, 1.0]
gdf_float['2'] = [4.0, 2.0, 1.0]

# Setup and fit clusters
dbscan_float = DBSCAN(eps=1.0, min_samples=1)
dbscan_float.fit(gdf_float)

print(dbscan_float.labels_)

输出:

0    0
1    1
2    2
dtype: int32

cuML还具有多GPU和多节点多GPU操作,使用Dask,用于越来越多的算法。 以下Python代码段从CSV文件中读取输入,并在单个节点上使用多个GPU在Dask工作器集群中执行NearestNeighbors查询:

# Create a Dask CUDA cluster w/ one worker per device
from dask_cuda import LocalCUDACluster
cluster = LocalCUDACluster()

# Read CSV file in parallel across workers
import dask_cudf
df = dask_cudf.read_csv("/path/to/csv")

# Fit a NearestNeighbors model and query it
from cuml.dask.neighbors import NearestNeighbors
nn = NearestNeighbors(n_neighbors = 10)
nn.fit(df)
neighbors = nn.kneighbors(df)

 

展开阅读全文

代码

的 Gitee 指数为
超过 的项目

cuML 的相关资讯

还没有任何资讯

cuML 的相关博客

强强联手 百度智能云机器学习平台BML上线NVIDIA RAPIDS加速库

近日,百度智能云与英伟达合作,通过容器化技术将RAPIDS库封装到机器学习平台BML,并利用Kubernetes进一步管理,做到了开箱即...

异构计算系列(二):机器学习领域涌现的异构加速技术

作者 | 易小萌、郭人通 策划 | 钰莹 “异构计算”(Heterogeneous computing),是指在系统中使用不同体系结构的处理器的联合...

异构计算系列(二):机器学习领域涌现的异构加速技术

作者 | 易小萌、郭人通 策划 | 钰莹 “异构计算”(Heterogeneous computing),是指在系统中使用不同体系结构的处理器的联合...

当 Mars 遇上 RAPIDS:用 GPU 以并行的方式加速数据科学

背景 在数据科学世界,Python 是一个不可忽视的存在,且有愈演愈烈之势。而其中主要的使用工具,包括 Numpy、Pandas 和 Scik...

使用Python玩转GPU

问题 随着机器学习对模型运算速度的需求越来越强烈, 一直想进行GPU编程,但一直以来这些都是c++的专利 一想到c++里的各种坑,...

模型训练竞品收集-百度

百度BML 一、产品介绍 百度机器学习BML(Baidu Machine Learning)是一款端到端的AI开发和部署平台。基于BML,用户可以一站式完...

【python量化】使用pyfinance进行证券收益分析

pyfinance简介 在查找如何使用Python实现滚动回归时,发现一个很有用的量化金融包——pyfinance。顾名思义,pyfinance是为投资...

【手把手教你Python】使用pyfinance进行证券收益分析

pyfinance简介 在查找如何使用Python实现滚动回归时,发现一个很有用的量化金融包——pyfinance。顾名思义,pyfinance是为投资...

技术债务的高息信用卡:深入了解那些知名的端到端机器学习平台

本文转载自公众号“读芯术”(ID:AI_Discovery)。 机器学习(ML)被称为技术债务的高利率信用卡。对于特定的业务问题,使用适用...

cuML 的相关问答

还没有任何问答,马上提问

评论 (0)

加载中
更多评论
0 评论
3 收藏
分享
返回顶部
顶部