当前位置: 首页 > 工具软件 > DLPack > 使用案例 >

基于Dlpack,Cupy.array和pytoch的tensor可相互转化

司马振国
2023-12-01

Cupy.array<-->Dlpack.Tensor<-->torch.Tensor

当前安装Cupy需要从源码安装,因为直接pip安装没有接口函数

cupy(左边和中间的转换)

from cupy.core.dlpack import toDlpack

from cupy.core.dlpack import fromDlpack

torch(中间和右边的转换)

from torch.utils.dlpack import to_dlpack

from torch.utils.dlpack import from_dlpack

实时性:从左到中间很快,但是从中间到左边很慢,因为需要先初始化一个torch.Tensor。

 类似资料: