3060筆記本顯卡
Win10 64位
問題:之前8個批次會爆顯存,只能跑6個批次。今天突然能跑到32個批次。求解是改了哪里,會影響這么大?
問題相關背景:
原環境: cuda11.1(驅動和運行時) +cudnn
然后用 conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia 安裝pytorch
5月初,pytorch 同樣網路,跑的batchsize 變為原來6倍。 之前“efficientB0”只能跑batchsize=6,現在最高可以到batchsize=32
更改:
這幾天就另外安裝tensorflow2.5 。 然后安裝cuda11.0(運行時) + 對應cudnn 。
然后發現原來的pyotch程式可以跑到 32個批次。
轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/284399.html
標籤:人工智能技術
下一篇:制作巖漿字體
