这个问题折磨了我很久,最后排查出来两个原因:
1)使用tf的dataloader,在map的时候后面多了一个“,”,在跑了几个的时候会提示你维度错误;
2)AUC错误、7209错误、不返回任何具体错误的错误,版本与cuda不符。比如tf10下无法训练的代码,在tf2.6下可以,但在CPU条件下两个代码都没有问题…
tensorflow在CPU可以正常运行,在GPU报错
最新推荐文章于 2023-02-20 16:37:27 发布
这个问题折磨了我很久,最后排查出来两个原因:
1)使用tf的dataloader,在map的时候后面多了一个“,”,在跑了几个的时候会提示你维度错误;
2)AUC错误、7209错误、不返回任何具体错误的错误,版本与cuda不符。比如tf10下无法训练的代码,在tf2.6下可以,但在CPU条件下两个代码都没有问题…