search

lxztju / pytorch_classification

利用pytorch实现图像分类的一个完整的代码，训练，预测，TTA，模型融合，模型部署，cnn提取特征，svm或者随机森林等进行分类，模型蒸馏，一个完整的代码
MIT License
1.38k stars 338 forks source link

关于多卡训练的问题 #43

Closed Maxhyl closed 2 years ago

Maxhyl commented 2 years ago

我用3个12G的卡训练，但是第一个卡显存会爆掉，但是第二和第三张卡还有很多显存，只有设置batchsize为1才不会报错，但是用单卡来跑的话batchsize可以设置为6，请问这个问题怎么解决可以用多卡且可以用较大的batchsize