search

potamides / AutomaTikZ

Text-Guided Synthesis of Scientific Vector Graphics with TikZ
Apache License 2.0
71 stars 3 forks source link

run examples/train.py with two 3090 GPUS? #9

Open AdjugateMatrix opened 1 year ago

AdjugateMatrix commented 1 year ago

How can I run the program with two GPUS? Do I need to add some code to support two GPUS? Hopefully you can give me some instructions. Thanks!

potamides commented 1 year ago

For training, multiple GPUs should be picked up automatically. Just execute examples/train.py or run it with torchrun --nproc_per_node gpu examples/train.py.