Closed aligoglos closed 5 years ago
I think it's torch issue in windows.
I think solution is to add torch.cuda.current_device()
before any other cuda calls, so you can add to this scripts after import torch
import torch
torch.cuda.current_device()
I tried to run
eval.py
but got this error:os : windows 10 python : 3.5.4 torch : 1.0.1