项目是\NLP经典项目集\10快递单信息抽取 里的demo
环境win10, wsl2 ,cuda11.2 paddle gpu 2.1
在win10里能运行,运行时显存占用2G左右
但是在wsl2里就只能运行前几步, 然后就不动了(控制台没有新输出), batch_size设置为2也不行
怀疑是Model.fit()的问题, 我就改用普通的 for step, batch in enumerate(train_loader, start=1) 这种来运行也不行(在win10上可以)
大佬们给个指点,谢谢!!
你wsl2里是有配置成功GPU环境吗？ 慢感觉可能是因为你是CPU训练。 检验下你paddle-gpu是否可用
paddle.utils.run_check()测试没问题, 而且创建的tensor也在CUDAPlace(0)上
其他的demo有的可以运行,比如\NLP打卡营\2文本相似度 可以在wsl2里运行
你wsl2里是有配置成功GPU环境吗？ 慢感觉可能是因为你是CPU训练。 检验下你paddle-gpu是否可用
paddle.utils.run_check()测试没问题, 而且创建的tensor也在CUDAPlace(0)上
其他的demo有的可以运行,比如\NLP打卡营\2文本相似度 可以在wsl2里运行