Suppose we run the following code on a GPU-enabled computer.
import tensorflow as tf
print ("results")
id= tf.eye(2,2)
x= tf.matmul(id,id) #line 4
print (x)
Will line #line 4 be executed on GPU automatically?
Asked
Active
Viewed 36 times
0
zell
- 9,830
- 10
- 62
- 115
-
https://stackoverflow.com/a/55608481/8143158 – Vijay Mariappan Nov 26 '22 at 05:53
-
In your link, "tf.test.is_gpu_available" asks if GPU is available. It does not mean GPU will be necessarily be used for tensor calculation I guess – zell Nov 26 '22 at 06:02
-
Oh, I see. Maybe you mean "tf.test.is_built_with_cuda()" means the code runs on GPU? – zell Nov 26 '22 at 06:06
-
TF defaults to gpu when available. You can also check variables where its executed by `x.device`. – Vijay Mariappan Nov 26 '22 at 06:32