I don’t think it’s a shape mismatch error, but an internal cudnn issue, which is why we would need to get a code snippet to reproduce it.
Also, did you install the pip wheels for 1.8.1? If so, could you create a new env and install the conda binaries?