在我们进行模型训练时,我们可能会疑惑代码什么的都已经进行了设置,到底在训练时是否用到了GPU训练呢?
我们可以尝试以下方法进行确定:
1.win+r 输入cmd
2.输入nvidia-smi
如果红色箭头为0%,表示没有用到gpu进行模型训练。
而下面这个就是用到了GPU训练。
本次学习交流就到此为止了,希望可以帮到你。
如何确定模型训练时是否使用了GPU训练
猜你喜欢
转载自blog.csdn.net/Kirihara_Yukiho/article/details/131549094
今日推荐
周排行