I’m at a loss. I just upgraded from pytorch 1.0 to pytorch 1.2 and see huge slowdowns in training (50%-70%). What could be causing this discrepancy? I’m building my docker container off of official nvidia/cuda:10.0 image and haven’t changed anything except upgrading pytorch.
The command for upgrading was:
conda install -c pytorch pytorch=1.2.0=py3.7_cuda10.0.130_cudnn7.6.2_0 torchvision
What am I missing?