We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 054ddae commit c86a8fbCopy full SHA for c86a8fb
src/backends/tensorflow.c
@@ -168,10 +168,10 @@ RAI_Graph *RAI_GraphCreateTF(const char *prefix,
168
// For setting config options in session from the C API see:
169
// https://github.com/tensorflow/tensorflow/issues/13853
170
// import tensorflow as tf
171
- // config = tf.ConfigProto()
172
- // config.intra_op_parallelism_threads = 1
+ // config = tf.ConfigProto(device_count = {'GPU': 0})
173
// serialized = config.SerializeToString()
174
// result = list(map(hex, serialized))
+ // print(result)
175
176
// TODO: complain if device is GPU and GPU not available?
177
if (device == RAI_DEVICE_CPU) {
0 commit comments