RuntimeError:cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/generic

RuntimeError:cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/generic/THCTensorMathPointwise.cu:324

In other people’s Python code, add their own module, CUDA run-time error. I’ve looked at the explanations of many bloggers. It’s been a long time without success
the main reason for this error is that the function in the added module can not be executed in GPU, and the incoming data is converted to CPU mode, and then converted to CUDA mode when it is called again after execution

Read More: