How to run multiple tensorflow codes in one gpu?

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

import tensorflow as tf

# Create a TensorFlow session
with tf.device('/gpu:0'):
    # Define and run the first TensorFlow code
    a = tf.constant([1.0, 2.0, 3.0])
    b = tf.constant([4.0, 5.0, 6.0])
    c = a + b

    with tf.Session() as sess:
        result1 = sess.run(c)
        print("Result 1:", result1)

# Create a TensorFlow session
with tf.device('/gpu:0'):
    # Define and run the second TensorFlow code
    x = tf.constant([7.0, 8.0, 9.0])
    y = tf.constant([10.0, 11.0, 12.0])
    z = x * y

    with tf.Session() as sess:
        result2 = sess.run(z)
        print("Result 2:", result2)