Still similar errors, tested in the following environments:
pytorch nightly v1.5, cudatoolkit 10.1, installed using conda
pytorch nightly v1.5, cudatoolkit 10.2, installed using conda
pytorch v1.4, cudatoolkit 10.1, installed using conda
pytorch v1.6, cudatoolkit 10.2, cudnn7.6.5, installed from source
number of successful epochs varied from 1-6