Can you add an option to use or not use the GPU when calling PyTorch?

Open shyamalschandra opened this issue 7 years ago • 1 comments

Otherwise, I will do it myself and check it in! Thanks for your time!

Mar 01 '18 16:03 shyamalschandra

There is already this option. Just add --num_gpu 0 when running main.py in the console.

Mar 02 '18 15:03 gregoire22enpc