[SOLVED] Titan V on PyTorch 0.3.0, CUDA 9.0, CUDNN 7.0 is much slower than 1080 Ti

[smth]

you also need this:

import torch
# after import torch, add this
torch.backends.cudnn.benchmark=True