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。