I’m using the linked codes without modification since I’m just testing a new machine.
After some test, I found:
cudatoolkit               10.2.89 + cudnn                     7.6.5 + pytorch                   1.5.1 fails;
cudatoolkit               10.0.130 + cudnn 7.4.2 + pytorch                   1.0.1 works.