The cuDNN library is designed for GPU-accelerated computations. If you encounter the error only when running on a GPU and not on CPU, it suggests that there may be GPU-specific issues such as memory overflow or compatibility problems.