Conflict between PyTorch version and CUDA version
Sinnerboy89 opened this issue · 1 comments
Sinnerboy89 commented
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 commented
Currently running a fork with torch 0.4.1 and cuda 9.1.