1. 程式人生 > >基於Dlpack,Cupy.array和pytoch的tensor可相互轉化

基於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。