PF-AFN icon indicating copy to clipboard operation
PF-AFN copied to clipboard

runtime error while training in the .sh file

Open ghost opened this issue 3 years ago • 2 comments

When im running "bash scripts/train_PBAFN_stage1.sh" its giving me the following error

RuntimeError: cuda runtime error (10) : invalid device ordinal at ..\torch\csrc\cuda\Module.cpp:33

device ordinal are:0 NVIDIA GeForce MX13

urgent help needed!!

ghost avatar Apr 20 '21 16:04 ghost

@Anaghadarshan Did you solve this problem?

kris-yangjs avatar May 14 '21 07:05 kris-yangjs

if you have just one gpu, open train_PBAFN_stage1.sh file and change --nproc_per_node=8 to --nproc_per_node=1

mafia1990 avatar May 24 '21 10:05 mafia1990