deeriny:

I doubt this is a cudnn version problem, and I want to change cudnn to 8.0.2 or 8.0.3.
How can I change only the cudnn version in pytorch?


I think it is indeed a cudnn problem, since you would be running into this issue 34 in the 1.8.0 release (so you should update to 1.9.0 or the nightly release) and also might want to build PyTorch from source using the latest cudnn release (8.2.1).