search

GeoX-Lab / LSTTN

29 stars 4 forks source link

RuntimeError: Device num not match, cfg.DEVICE_NUM = 2, but torch.cuda.device_count() = 0 #7

Open ThisAngryman opened 2 months ago

ThisAngryman commented 2 months ago

RuntimeError: Device num not match, cfg.DEVICE_NUM = 2, but torch.cuda.device_count() = 0 有人遇到这个问题吗？我用的torch版本是1.8.1gpu的但还是报这个错误

1zhd commented 1 month ago

楼主您好，我现在复现时将launch_training(CFG)放在main的前两行也会报这个错误；请问楼主有解决这个问题，期待楼主回复，谢谢！