博文
Pytorch training with multi GPUs
||
When using DataParallel, with a list of gpu like [dev0, dev1, ...], all the inputs that you give to the module have to be on dev0. We can use the following codes to set
with torch.cuda.device(dev0): t = t.cuda()
https://blog.sciencenet.cn/blog-1969089-1181522.html
上一篇：3D ResNet pre-trained on Kinetics
下一篇：Python图中中文字体设置
