CODEWITHSUNDEEP
amazon-web-serviceskerasgpu
Mauro Gentile
Mauro Gentile

Reputation: 1511

How do I check if I am using GPU on AWS?

I am training a NN using keras in a deep learning AMI (ubuntu 12) in AWS.

How do I ensure that my script in Jupyter Notebook is using GPU?

Upvotes: 0

Views: 541

Answers (1)

Richard Wan
Richard Wan

Reputation: 34

Before running the session in tensorflow you could add this line of code to check:

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

For further detail of using GPU with tensorflow, you could check it out on tensorflow official website: https://www.tensorflow.org/guide/using_gpu

Upvotes: 1

Related Questions