问题描述:

1、使用context将模型放到GPU上,执行训练时,模型使用CPU进行计算。

 

【截图信息】

解答:

已经使用了GPU,看显存的占用,且perf性能等级已经是p2了,说明已经使用了GPU,只是你的代码GPU利用率比较低,可能自定义的训练过程里面CPU的执行成分比较多,导致看上去似乎没使用GPU

更多推荐