I want to transfer my data preprocession into GPU, and I have writen a series of code to do something to a torch.cuda.tensor before feeding it into network, the code was proved to be true. But when I actually use this way to train my model, the cuda runtime initialization error always occure. I can’t understand where the problem is.
The error massage is
RuntimeError: cuda runtime error (3) : initialization error at /home/ayu/pytorch/aten/src/THC/THCCachingAllocator.cpp:508
Before I fed data to model, the defined model has been transfered to GPU, I don’t know if this is the problem.
Thank you for your advice!