search

zju3dv / EasyVolcap

[SIGGRAPH Asia 2023 (Technical Communications)] EasyVolcap: Accelerating Neural Volumetric Video Research
Other
577 stars 41 forks source link

如何支持多卡训练 #27

Closed prettybot closed 4 months ago

prettybot commented 4 months ago

作者您好，这里提到可以多卡训练，具体应该怎么设置呢 https://github.com/zju3dv/EasyVolcap/issues/21#issuecomment-1912105861

dendenxu commented 4 months ago

你好，只需将evc替换成evc-dist即可多卡训练。会根据当前的CUDA_VISIBLE_DEVICES选择训练用卡。

prettybot commented 4 months ago

非常感谢！