Commit 7bfa87ea authored by Akshay Agrawal's avatar Akshay Agrawal
Browse files

minor fix

parent 554bb86f
......@@ -282,7 +282,7 @@
},
"cell_type": "code",
"source": [
"if tf.test.is_gpu_available() > 0:\n",
"if tf.test.is_gpu_available():\n",
" with tf.device(tf.test.gpu_device_name()):\n",
" B = tf.constant([[2.0, 0.0], [0.0, 3.0]])\n",
" print(tf.matmul(B, B))"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment