hmmm, that’s interesting, then maybe I have a cuda or cudnn issue
the exact torch and pytorch versions are listed in the question

pytorch                   1.9.0           py3.8_cuda10.2_cudnn7.6.5_0    pytorch