Skip to content

Commit 8e9b536

Browse files
sboshinnadiaya
authored andcommitted
fix: fixing dockerfile bug, && on both end and beginning of the line (#256)
* Fixing dockerfile bug, && on both end and beginning of the line * Fixing gpu placement to not use session
1 parent 59f3c47 commit 8e9b536

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/resources/gpu_device_placement.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,12 @@
1515
# https://www.tensorflow.org/programmers_guide/using_gpu
1616
print('-' * 87)
1717
print('Run GPU test.')
18+
tf.debugging.set_log_device_placement(True)
1819
with tf.device('/gpu:0'):
1920
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
2021
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
2122
c = tf.matmul(a, b)
22-
sess = tf.compat.v1.Session(config=tf.compat.v1.ConfigProto(log_device_placement=True))
2323
# Runs the op.
24-
print(sess.run(c))
24+
print(c)
2525
print('-' * 87)
2626
print('')

0 commit comments

Comments
 (0)