PDVC icon indicating copy to clipboard operation
PDVC copied to clipboard

Published 20 hours ago • verified publisher icon ttengwang

How to use 2 GPUs?

Open yhd-123 opened this issue 1 year ago • 2 comments

image python train.py --cfg_path ${cfgs/anet_c3d_pdvc.yml} But only one GPU is used!

yhd-123 avatar Jan 09 '24 10:01 yhd-123

This repository does not currently support data parallelism for multiple gpus. If you are interested, you can modify the code accordingly. A helpful reference is https://pytorch.org/tutorials/intermediate/model_parallel_tutorial.html

ttengwang avatar Jan 11 '24 17:01 ttengwang

OK！Thanks!

yhd-123 avatar Jan 12 '24 06:01 yhd-123