zi2zi-pytorch icon indicating copy to clipboard operation
zi2zi-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon EuphoriaYan

在第三部训练时出现AssertionError

Open doublex640 opened this issue 1 year ago • 2 comments

Traceback (most recent call last): File "train.py", line 128, in main() File "train.py", line 79, in main model.setup() File "F:\zi2zi-pytorch-master\model\model.py", line 59, in setup init_net(self.netG, gpu_ids=self.gpu_ids) File "F:\zi2zi-pytorch-master\utils\init_net.py", line 15, in init_net assert(torch.cuda.is_available()) AssertionError

doublex640 avatar Dec 27 '23 01:12 doublex640

我的显卡是RTX660，可以支持cuda10.2

doublex640 avatar Dec 27 '23 13:12 doublex640

问题出在cuda没有被调用，环境重新配置了一遍解决了

doublex640 avatar Jan 03 '24 13:01 doublex640