The error is
“THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=405 error=11 : invalid argument”. But it doesn’t influence the training and test, I want to know the reason for this error.
My cuda version is 9.0 and the python version is  3.6.

image.png698×232 5.59 KB

Thank you for help