Question: How I can use TF + nGraph + PlaidML ?

NervanaSystems / ngraph-tf

Bridge to connect nGraph with TensorFlow

Other

53 stars 16 forks source link

Question: How I can use TF + nGraph + PlaidML ? #534

Open jaebaek opened 5 years ago

jaebaek commented 5 years ago

I installed TF and nGraph using python3 build_ngtf.py --build_plaidml_backend --use_prebuilt_tensorflow

I tested TF mnist hello world. It uses only CPU. How can I use PlaidML + OpenCL for its backend?

cepera commented 5 years ago

hi, you need to set NGRAPH_TF_BACKEND env variable to PLAIDML. export NGRAPH_TF_BACKEND="PLAIDML"

SleepProgger commented 5 years ago

@cepera Did you get it to run with plaidml ? When i try i get an tensorflow.python.framework.errors_impl.InternalError: Caught exception while compiling op_backend: get_shape() must be called on a node with exactly one output () error or a segfault (random)

cepera commented 5 years ago

hi @SleepProgger, well, I did run it with plaidml like... in March, it worked fine for me. I would recommend to create a new issue here with some steps to reproduce