Segmentation fault (core dumped)问题解决

torch:1.10.0+cu111
pandas:1.3.3

报错:

Segmentation fault (core dumped)问题解决_第1张图片

 定位:

df = pd.DataFrame(columns=[epoch, model.sigma1.cpu().detach().numpy(), model.sigma2.cpu().detach().numpy(),model.sigma3.cpu().detach().numpy(),iter,avg_ssim, avg_psnr,model.current_log()])

解决方法:pandas升级至1.4.4以上

detach是把torch计算过程中数据里的梯度取消,这样可以用dataframe方法写入csv,如果不用detach会提示,但是用了就会终止,生成core.xx,Segmentation fault 问题经常是内存溢出,但实际查看内存没有超过,就升级了pandas,然后问题解决。

你可能感兴趣的:(python,人工智能)