訓練神經網絡時,batchsize的設置方法,以達到GPU算力的充分利用

先在窗口轉路徑 cd C:\Program Files\NVIDIA Corporation\NVSMI   而後輸入指令nvidia -smi 就會顯示圖右側的信息框,那裏能夠看到顯卡總顯存和正在佔用的顯存數值。網絡 此外,更直接的方法就是到任務管理器 -> 性能 -> GPU0 -> 專用GPU內存,就能夠看到總顯存及實時佔用顯存。性能 注:batchsize大小與網絡複雜程度負相關,也即不一
相關文章
相關標籤/搜索