Uploaded image for project: 'Beam'
  1. Beam
  2. BEAM-13970 RunInference V1
  3. BEAM-13986

Provide GPU support

    XMLWordPrintableJSON

Details

    • Sub-task
    • Status: Open
    • P2
    • Resolution: Unresolved
    • None
    • None
    • sdk-py-core

    Description

      Pytorch and Tensorflow have GPU trained models. Need to support this in the RunInference classes, and also make sure they get configured in Dataflow.

      Attachments

        Activity

          People

            Unassigned Unassigned
            yeandy Andy Ye
            Votes:
            0 Vote for this issue
            Watchers:
            2 Start watching this issue

            Dates

              Created:
              Updated: