WEAN icon indicating copy to clipboard operation
WEAN copied to clipboard

Executing over CPU

Open surajpadhy opened this issue 5 years ago • 2 comments

Hello,

In list of parameters i see only GPU option, is there any way to train the model using CPU? My laptop has only Intel's inbuild graphics memory available (Graphics: Intel® Kabylake GT1.5). Getting error:

RuntimeError: cuda runtime error (35) : CUDA driver version is insufficient for CUDA runtime version at torch/csrc/cuda/Module.cpp:32 [INFO/MainProcess] process shutting down

How do i use it? Any assistance is highly appreciated.

surajpadhy avatar Aug 20 '19 08:08 surajpadhy

how about to set gpu parameter to -1?

superwoou avatar Nov 21 '19 06:11 superwoou

Add any negative integer value in the args. Eg: python3 run.py -gpu -2 -mode evaluate -restore save_path/best.th

gdgovinddaga6 avatar Jan 09 '20 10:01 gdgovinddaga6