kris-yangjs
kris-yangjs
@geyuying OK, I'll try, thanks for your reply
> > 嗨,我的環境也是1個GPU。您是否成功運行了培訓文件?例如train_PBAFN_stage1.py? > > not , > -------------- End ---------------- > THCudaCheck FAIL file=/pytorch/torch/csrc/cuda/Module.cpp line=33 error=10 : invalid device ordinal > Traceback (most recent call last): > File "train_PBAFN_stage1.py",...
@Anaghadarshan Did you solve this problem?
I hava the same question while using pytorch1.8.0.
@lyx0208 Thanks, I'll try it later
@lyx0208 Hava you ever got this error? 