PaddleHub 如何提升训练的速度

rur96b6h  于 5个月前  发布在  其他
关注(0)|答案(2)|浏览(85)

请问如何提升训练的速度,望能提供一些好的建议,比如CPU和GPU利用率等?

ilmyapht

ilmyapht1#

可以尝试多卡训练,设置环境变量CUDA_VISIBLE_DEVICES=0,1,2,3,然后执行train.py脚本。

ilmyapht

ilmyapht2#

可以尝试多卡训练,使用以下命令:

CUDA_VISIBLE_DEVICES=0,1,2,3 python -m paddle.distributed.launch train.py 执行train.py脚本。

关于paddle.distributed.launch,你可以参考PaddlePaddle官方文档中的分布式训练部分。

相关问题