Why can't train the model on the GPU
JasonChenhx opened this issue · 3 comments
I don't modify the code about GPU, and "CUDA is available. Training on GPU".
What's more, I have put the input and model into the cuda. So What else do I need to put in cuda?
Thanks.
What does the output to 'device' give you?
If it's cuda:0 then its a GPU, otherwise it's being trained on CPU.
Other than that, everything is there for training on a GPU or a CPU, according to the device found.
What does the output to 'device' give you?
If it's cuda:0 then its a GPU, otherwise it's being trained on CPU.Other than that, everything is there for training on a GPU or a CPU, according to the device found.
The run result is cuda:0, and CUDA is available. Training on GPU"
And I have put the input、target、model into the cuda.
They all will be trained in the GPU and the will be transferred back to the CPU.
Look into the line 244 in pytorch_run
It transfers your training data to GPU and then as you see in line 298 it will be called back to the CPU.