pytorch调试过程中问题总结(持续更新)

1 pkl文件读取问题 问题代码:

with open('data.pkl', 'rb') as fp: sample_datas = pickle.load(fp)

错误信息: UnicodeDecodeError: ‘ascii’ codec can’t decode byte 0xc7 in position 1: ordinal not in range(128) 解决方法:

with open('data.pkl', 'rb') as fp: sample_datas = pickle.load(fp, encoding='iso-8859-1')

2 CUDNN在模型训练过程中报错 问题描述: 有时候在pytorch训练模型过程中CUDNN会提示报错.例如:

net = ResNet101() net = net.cuda() cudnn.benchmark = True net.train()

错误信息: RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED 解决方法: 此种方法可在网络后面添加torch.backends.cudnn.enabled=False,如:

net = ResNet101() net = net.cuda() cudnn.benchmark = True torch.backends.cudnn.enabled=False net.train()

自己在学习中记录总结问题,也方便以后遇见类似问题,快速找到解决方法,遇见新的问题逐步更新~~~