tjk:

We tried to run the same program on CUDA 11.1 and pytorch 1.8.1, it can run three epoches, and the error :RuntimeError: CUDA error: an illegal memory access was encountered would pop out.


Since you are seeing the cuDNN issue apparently in both setups, could you post an executable code snippet which would reproduce the illegal memory access on the A6000, please?