menu
SkyTowner
BETA
search
Search
Login
Map of Data Science
menu
menu
SkyTowner
search toc more_vert
Robocat
Guest 0reps
Sign up
Log in
account_circleMy Profile homeAbout paidPricing
emailContact us
exit_to_appLog out
Map of data science
Thanks for the thanks!
close
history_edu
Blog
 help
Help
Outline
Solution
chevron_left TensorFlow
Checking if GPU is available on TensorFlow 2Introduction to TensorFlow
Comments
Log in or sign up
Cancel
Post
account_circle
Profile
exit_to_app
Sign out
help Ask a question
Share on Twitter
search
keyboard_voice
close
Searching Tips
Search for a recipe:
"Creating a table in MySQL"
Search for an API documentation: "@append"
Search for code: "!dataframe"
Apply a tag filter: "#python"
Useful Shortcuts
/ to open search panel
Esc to close search panel
↑↓ to navigate between search results
⌘d to clear all current filters
⌘Enter to expand content preview
icon_star
Doc Search
icon_star
Code Search Beta
SORRY NOTHING FOUND!
mic
Start speaking...
Voice search is only supported in Safari and Chrome.
fullscreen_exit
Shrink
north_east
Navigate to
A
A
brightness_medium
share
arrow_backShare
Twitter
Facebook
chevron_left TensorFlow
Checking if GPU is available on TensorFlow 2Introduction to TensorFlow
check_circle
Mark as learned
thumb_up
0
thumb_down
0
chat_bubble_outline
0
auto_stories new
settings

Checking if GPU is available on TensorFlow 2

Machine Learning
chevron_right
TensorFlow
schedule Jul 1, 2022
Last updated
local_offer
Tags
tocTable of Contents
expand_more
Solution
map
Check out the interactive map of data science

Solution

To check if GPU is available on TensorFlow:

import tensorflow as tf

is_gpu_available = (len(tf.config.list_physical_devices('GPU')) > 0)
print(is_gpu_available)
True

Here, we are first obtaining the list of available GPUs like so:

tf.config.list_physical_devices('GPU')
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]

If a GPU is not available, then you will get an empty list.

robocat
Published by Isshin Inada
Edited by 0 others
Did you find this page useful?
thumb_up
thumb_down
Comment
Citation
Share on Twitter
Share on Facebook
Ask a question or leave a feedback...
thumb_up
0
thumb_down
0
chat_bubble_outline
0
settings
Enjoy our search
Hit / to insta-search docs and recipes!
Navigation
Home Blog Publish Help
Contact us
Resources
Python Pandas MySQL Beautiful Soup Matplotlib NumPy PySpark Machine Learning
Community
Join our Discord
 mail
Join our newsletter for updates on new comprehensive DS/ML guides
Terms
|
Privacy