Open aigerimmmm opened 2 years ago
metoo
How did you install pytorch? It sounds like you installed pytorch without CUDA support. https://pytorch.org/ has instructions for how to install pytorch with cuda support.
@aigerimmmm try this setting --GPU -1
Try running on CPU by adding the following option: --gpu_ids -1
@aigerimmmm try this setting
--GPU -1
This solution solved the problem
But it seems that it is not using the GPU, but using the CPU.
torch._C._cuda_setDevice(device) AttributeError: module 'torch._C' has no attribute '_cuda_setDevice' Finish Stage 1 ...