加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
start.sh 439 Bytes
一键复制 编辑 原始数据 按行查看 历史
纵横 提交于 2020-06-27 19:50 . Fix seed bug and update readme
python multiprocessing_distributed.py
CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --nproc_per_node=4 distributed.py
CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --nproc_per_node=4 apex_distributed.py
HOROVOD_WITH_PYTORCH=1 CUDA_VISIBLE_DEVICES=0,1,2,3 horovodrun -np 4 -H localhost:4 --verbose python horovod_distributed.py
srun -N2 --gres gpu:4 python distributed_slurm_main.py --dist-file dist_file
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化