如何确定模型训练时是否使用了GPU训练

在我们进行模型训练时,我们可能会疑惑代码什么的都已经进行了设置,到底在训练时是否用到了GPU训练呢?
我们可以尝试以下方法进行确定:
1.win+r 输入cmd
在这里插入图片描述
2.输入nvidia-smi在这里插入图片描述
如果红色箭头为0%,表示没有用到gpu进行模型训练。
而下面这个就是用到了GPU训练。
在这里插入图片描述本次学习交流就到此为止了,希望可以帮到你。

猜你喜欢

转载自blog.csdn.net/Kirihara_Yukiho/article/details/131549094