[
https://issues.apache.org/jira/browse/FLINK-22205?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=17319038#comment-17319038
]
Yangze Guo edited comment on FLINK-22205 at 4/12/21, 5:58 AM:
--------------------------------------------------------------
I have a [demo|https://github.com/KarmaGYZ/flink-mnist] which running inference
task with GPU. But it does not rely on TensorFlow. Maybe
[this|https://github.com/alibaba/flink-ai-extended/tree/master/deep-learning-on-flink/flink-ml-examples/src/main/java/com/alibaba/flink/ml/examples/tensorflow/mnist]
is what you need. Can [~zhongwei] make sure does it support GPU version
TensorFlow as well?
cc [~jqin]
was (Author: karmagyz):
I have a [demo|https://github.com/KarmaGYZ/flink-mnist] which running inference
task with GPU. But it does not rely on TensorFlow. Maybe
[this|https://github.com/alibaba/flink-ai-extended/tree/master/deep-learning-on-flink/flink-ml-examples/src/main/java/com/alibaba/flink/ml/examples/tensorflow/mnist]
is what you need. Can [~weizhong] make sure does it support GPU version
TensorFlow as well?
cc [~jqin]
> Does it support gpu coding in flink?
> ------------------------------------
>
> Key: FLINK-22205
> URL: https://issues.apache.org/jira/browse/FLINK-22205
> Project: Flink
> Issue Type: Task
> Components: API / Core
> Affects Versions: 1.12.0
> Reporter: ying zhang
> Priority: Major
>
> HI,I am running a tf inference task on my cluster,but I flind it took so long
> a time to get response, becase it is a bert model and I run it on cpu
> machine.My componey has gpu k8s cluster,and I read the document
> [https://ci.apache.org/projects/flink/flink-docs-master/docs/deployment/advanced/external_resources/|https://ci.apache.org/projects/flink/flink-docs-master/docs/deployment/advanced/external_resources/,but]
>
> count you give me a demo?Including tf inference on gpu and train on gpu?
> I use alink in some of my task, is there a demo for alink on gpu?
--
This message was sent by Atlassian Jira
(v8.3.4#803005)
