Documentation Center

GPU configuration

Procedure

  • Translation engines

    Each translation engine is configured independently. In order to turn the GPU support on, use gpu: true flag in configuration.

    mtedge:
      ...
      tengines:
      ...
        engfra:
          lp: engfra_generic_srv_tnmv_8_4_x_1
          gpu: true
  • Training engines

    The GPU support applies to all the training engines. In order to turn GPU support on, use gpu: true flag in configuration.

    mtedge:
      ...
      trainingEngines:
        replicas: 1
        gpu: true