在训练神经网络模型时,如果训练的模型占用的显存较小
如果需要多次调参,在单线程下,只能等待程序跑完,然后再修改参数,再运行,即浪费了时间,GPU的资源也没有很好利用。这时候可以开多线程,每个线程跑不同的超参,即提高的GPU的利用率,也节省了时间。
具体 *** 作如下:在运行按钮
处选择Edit Configurations...
,选择Allow parallel run
,保存即可.
输入nvidia-smi
查看显卡使用情况,可以看到有多个线程同时在运行
到有多个线程同时在运行
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)