• 博客园logo
  • 会员
  • 众包
  • 新闻
  • 博问
  • 闪存
  • 赞助商
  • HarmonyOS
  • Chat2DB
    • 搜索
      所有博客
    • 搜索
      当前博客
  • 写随笔 我的博客 短消息 简洁模式
    用户头像
    我的博客 我的园子 账号设置 会员中心 简洁模式 ... 退出登录
    注册 登录
华东 博客
17年国科大博士毕业,曾就职于三星电子,清华博后,目前在某大模型创业公司工作,研究方向大模型、智能体 新浪博客: http://blog.sina.com.cn/u/2463286753
博客园    首页    新随笔    联系   管理    订阅  订阅
2023年1月7日
nvidia-rapids (cuML)机器学习加速库安装和使用
摘要: 背景: sklearn是非常成熟、使用方便的机器学习算法工具包,但是其运行速度慢,只能支持CPU运行, 一直是一个诟病。 现在也出现了对sklearn中算法加速的工具包,比如thundersvm、sklearnex, 但是测试之后,加速效果十分有限。 下面提供NVIDIA开发基于GPU的cuML机器 阅读全文
posted @ 2023-01-07 20:50 华东博客 阅读(10995) 评论(0) 推荐(0)
采用faiss-gpu实现比Sklearn中K紧邻分类器(K-NN)100倍以上加速
摘要: 最近使用sklearn跑一些机器学习的实验对比,发现许多算法随着数据集增大,训练时间呈几何增加,加之交叉验证、参数选择等,非常耗时。 对此,已经有许多优化方案被提出。这里给出一个关于K-NN分类算法的快速实现工具推荐:基于faiss实现版本, 亲测速度提升明显 (1) 安装GPU版本faiss (h 阅读全文
posted @ 2023-01-07 15:49 华东博客 阅读(1351) 评论(0) 推荐(0)
博客园  ©  2004-2025
浙公网安备 33010602011771号 浙ICP备2021040463号-3