RANK_SIZE

Description

Sets the number of devices corresponding to the current training process in TensorFlow distributed training or inference scenarios.

Example

export RANK_SIZE=2

Constraints

None

Applicability

Atlas 350 Accelerator Card

Atlas A3 training product/Atlas A3 inference product

Atlas A2 training product/Atlas A2 inference product

Atlas training product

Atlas 300I Duo Inference Card