Update tensorflow.rst

jade-hpc-gpu · Nov 8, 2021 · 7a21748 · 7a21748
1 parent 0e67cbc
commit 7a21748
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/software/machine-learning/tensorflow.rst b/software/machine-learning/tensorflow.rst
@@ -70,8 +70,8 @@ Run the ``tftest.py`` script with the following command: ::
 
 Which gives the following results: ::
 
-tf.Tensor(
-[[22. 28.] [49. 64.]], shape=(2, 2), dtype=float32)
+  tf.Tensor(
+  [[22. 28.] [49. 64.]], shape=(2, 2), dtype=float32)
 
 
 Using Tensorflow in Batch Mode
@@ -145,8 +145,7 @@ The output will appear in the slurm standard output file with the corresponding
 
   NOTE: Legacy NVIDIA Driver detected.  Compatibility mode ENABLED.
 
-  [[ 22.  28.]
-	 [ 49.  64.]]
+  [[ 22.  28.][ 49.  64.]]
 
 
 Using multiple GPUs
@@ -175,5 +174,6 @@ You will see something similar to the following output. ::
 	I tensorflow/core/common_runtime/gpu/gpu_device.cc:1428] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:1 with 31031 MB memory) -> physical GPU (device: 1, name: Tesla V100-SXM2-32GB-LS, pci bus id: 0000:07:00.0, compute capability: 7.0)
 	I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
 tf.Tensor(
-[[ 44.  56.][ 98. 128.]], shape=(2, 2), dtype=float32)
+
+        [[ 44.  56.][ 98. 128.]], shape=(2, 2), dtype=float32)