search

XinyiYing / BasicIRSTD

BasicIRSTD toolbox
154 stars 24 forks source link

多卡训练 #29

Open QYR123 opened 4 days ago

QYR123 commented 4 days ago

你好，怎么进行多卡训练

XinyiYing commented 4 days ago

终端运行以下命令： CUDA_VISIBLE_DEVICES=0,1 python train.py

QYR123 commented 2 days ago

运行了还是单卡训练

XinyiYing commented 2 days ago

下面这句话指示了多卡训练， 1727922051944 我试了我这边可以多卡训练，可能因为model占的显存太小看不出来，可以加大batch size再训练看看不同显卡的显存占用。命令CUDA_VISIBLE_DEVICES=0,1 python train.py中的0,1指示的是使用的GPU的号码。