microsoft / Bringing-Old-Photos-Back-to-Life

Bringing Old Photo Back to Life (CVPR 2020 oral)
https://arxiv.org/abs/2004.09484
MIT License
15.14k stars 2k forks source link

AttributeError: module 'torch._C' has no attribute '_cuda_setDevice' #241

Open aigerimmmm opened 2 years ago

aigerimmmm commented 2 years ago

torch._C._cuda_setDevice(device) AttributeError: module 'torch._C' has no attribute '_cuda_setDevice' Finish Stage 1 ...

morelife5831 commented 2 years ago

metoo

mohamaddkhalill commented 2 years ago

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.

xiaoxinmiao commented 1 year ago

@aigerimmmm try this setting --GPU -1

wangshirely commented 1 year ago

Try running on CPU by adding the following option: --gpu_ids -1

alphapro-pro commented 1 year ago

@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.