Thanks for the kind reply. My workstation only has 4GB GPU memory. That is the problem.
I wonder if the data can be read and processed in GPU piece by piece. So not all the data is needed to be imported into the GPU memory. Is this function available in pytorch?
Thanks a lot.