how to train on cpu - Githubissues

polarisZhao / PFLD-pytorch

PFLD pytorch Implementation

808 stars 198 forks source link

how to train on cpu #13

Closed sljlp closed 4 years ago

sljlp commented 5 years ago

hi! I have no nvidia gpu on my computer .When I run the script train.py the error shows: "AssertionError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from" How can I train on cpu?

ideask commented 5 years ago

Maybe u can't train on cpu, because it takes about 8G graphics memory.