transformers 段错误 (核心已转储)

transformers在cuda8下的 BertForMaskedLM.from_pretrained 执行时会报 “段错误 (核心已转储)” 的错误,应该是cuda版本的问题,使用cuda10后错误消失

你可能感兴趣的:(深度学习框架,cuda)