CuPy

CuPy

开发框架/库 免费
NumPy/SciPy 兼容的 GPU 加速数组库,由 Chainer 团队开发
访问官网

搭配玩法

1
CuPy + PyTorch PyTorch + cuda.core

内核直接作用于 CuPy 数组或 PyTorch 张量,无需复制数据

cuda.core 使用标准 CUDA 上下文,其他组件可共享设备、流和内存

✓ 无需数据拷贝✓ 共享设备和流

产品简介

NumPy/SciPy 兼容的 GPU 加速数组库,由 Chainer 团队开发

用户评价

还没有评价
这款产品刚收录,成为第一个评价的人
写第一条评价