RuntimeError:CUDNN_STATUS_EXECUTION_FAILED

pytorch0.4,cuda9.0.176,cudnn7.0

Could you check, if your code is successfully running on CPU, and if not, post the error message?

When the training data is reduced to only 1000 , using CUDA can run successfully, but there will be problems when using all 100k data, which makes me very confused.