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

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。

你可能感兴趣的:(python,torch,cupy)