Conflict between PyTorch version and CUDA version

[Sinnerboy89]

Is it possible to run train.py with CUDA 9+?

train.py attempts to import torch.utils.trainer, which seems to have been removed from PyTorch at around version 1.0.0. However, I think 1.0.0 or newer is required for it to run on GPUs with CUDA 9+.

What's the easiest way to resolve this?

[dmtrs]

Currently running a fork with torch 0.4.1 and cuda 9.1.