[
https://issues.apache.org/jira/browse/BEAM-13986?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=17549866#comment-17549866
]
Danny McCormick commented on BEAM-13986:
----------------------------------------
This issue has been migrated to https://github.com/apache/beam/issues/21438
> Provide GPU support
> -------------------
>
> Key: BEAM-13986
> URL: https://issues.apache.org/jira/browse/BEAM-13986
> Project: Beam
> Issue Type: Sub-task
> Components: sdk-py-core
> Reporter: Andy Ye
> Priority: P2
> Labels: run-inference
>
> Pytorch and Tensorflow have GPU trained models. Need to support this in the
> RunInference classes, and also make sure they get configured in Dataflow.
--
This message was sent by Atlassian Jira
(v8.20.7#820007)
