Paddle 如何使用多块GPU 并行训练

来自官方的 教程

  #单机单卡启动,默认使用第0号卡
  $ python train.py
  
  #单机多卡启动,默认使用当前可见的所有卡
  $ python -m paddle.distributed.launch train.py

  #单机多卡启动,设置当前使用的第0号和第1号卡
  $ python -m paddle.distributed.launch --selected_gpus='0,1' train.py

  #单机多卡启动,设置当前使用第0号和第1号卡
  $ export CUDA_VISIABLE_DEVICES='0,1'
  $ python -m paddle.distributed.launch train.py

你可能感兴趣的:(Paddle,paddle,paddlepaddle)