GPU不参与训练 请问配置完对应GPU所需的cuda和需要的镜像源 在进行模型训练 CPU跑到100% GPU没进行运算 请问这个问题如何解决

[mochaYJ12138]

Summary[问题简述（一句话）] A clear and concise description of what the issue is.

Env & To Reproduce[复现与环境] 描述你用的环境、代码版本、模型

Screenshots[截图（如有）] If applicable, add screenshots to help

[HexBanana]

检查一下你的pytorch版本，pytorch如果是cpu版本就改成cuda版本，其他issue里也有类似的问题，可以看一下和你遇到的问题符不符合。我也遇到了这个问题，网上搜是最新版的bug，我选择的办法是降级，用pytorch1.11配cuda11.3，希望能帮到你