Error in pytorch1.6 training:
RuntimeError: CUDA error: an illegal memory access was encountered
The reason for the error is the same as that of the lower version of python (such as version 1.1)
Runtimeerror: expected object of backend CUDA but get backend CPU for argument https://blog.csdn.net/weixin_ 44414948/article/details/109783988
Cause of error:
The essence of this kind of error reporting is model and input data_ image、input_ Label) is not all moved to GPU (CUDA).
* * tips: * * when debugging, you must carefully check whether every input variable and network model have been moved to the GPU. I usually report an error because I have missed one or two of them.
resolvent:
Model, input_ image、input_ The example code is as follows:
model = model.cuda()
input_image = input_iamge.cuda()
input_label = input_label.cuda()
or
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model = model.to(device)
input_image = input_iamge.to(device)
input_label = input_label.to(device)