How to run Pytorch gpu tensor using matlab gpu?


(Gaurav Nayak) #1

I am running the code in pytorch using gpu which produces a matrix that is fed to the matlab code where the other logic is implemented.

The code is working fine but takes 10 to 15 min for one epoch using 2 gpu’s. The probable reason might be the data transfer from pytorch cuda tensor to matlab variable and then running on matlab in cpu and from matlab to pytorch tensor.

I am curious to know that is there any way such that I can use the pytorch tensor as it is on matlab code and force the matlab to run gpu with that pytorch cuda tensor.

Any help is highly appreciated. :slightly_smiling_face: