ARCC-RACE / deepracer-for-dummies

a quick way to get up and running with local deepracer training environment
66 stars 28 forks source link

CPU Training #29

Open caelinsutch opened 5 years ago

caelinsutch commented 5 years ago

Add support for CPU training or notify user if their GPU isn't powerful enough.

Michael-Equi commented 5 years ago

Support already exists. We could add a button to automatically enable it in the GUI.

Michael-Equi commented 5 years ago

You can set the environment variable GPU_AVAILABLE in the docker/.env file to false.

On Sun, Aug 18, 2019 at 9:12 PM Alex Feng notifications@github.com wrote:

@Michael-Equi https://github.com/Michael-Equi How can I do CPU training? I don't have a GPU.

— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub https://github.com/ARCC-RACE/deepracer-for-dummies/issues/29?email_source=notifications&email_token=AH3V2SW6AGWZ4FMZP2KLRW3QFIMRBA5CNFSM4IKGB6NKYY3PNVWWK3TUL52HS4DFVREXG43VMVBW63LNMVXHJKTDN5WW2ZLOORPWSZGOD4RUGIA#issuecomment-522404640, or mute the thread https://github.com/notifications/unsubscribe-auth/AH3V2SU2ZMKOF2U4OFCZ7CLQFIMRBANCNFSM4IKGB6NA .

shimin-happy commented 4 years ago

You can set the environment variable GPU_AVAILABLE in the docker/.env file to false. On Sun, Aug 18, 2019 at 9:12 PM Alex Feng @.***> wrote: @Michael-Equi https://github.com/Michael-Equi How can I do CPU training? I don't have a GPU. — You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub <#29?email_source=notifications&email_token=AH3V2SW6AGWZ4FMZP2KLRW3QFIMRBA5CNFSM4IKGB6NKYY3PNVWWK3TUL52HS4DFVREXG43VMVBW63LNMVXHJKTDN5WW2ZLOORPWSZGOD4RUGIA#issuecomment-522404640>, or mute the thread https://github.com/notifications/unsubscribe-auth/AH3V2SU2ZMKOF2U4OFCZ7CLQFIMRBANCNFSM4IKGB6NA .

I have set GPU_AVAILABLE is false, but still get OOM error. It looks like the training process still use GPU instead of CPU. Do you start training using ./start.sh script?