pytorch多卡训练记录

1.shell管道中运行:

python -m parallel_wavegan.distributed.launch --nproc_per_node 2 -c parallel-wavegan-train

此处-c 指需要多卡训练的任务命令

2.命令行运行:

python -m launch --nproc_per_node 2 train.py --canshu XXX

python launch.py --nproc_per_node 2 train.py --canshu XXX

此处不需要指定 -c

你可能感兴趣的:(大数据,pytorch,python)