Tensorflow gpu test

Tensorflow check gpu

tf.config.list_physical_devices('GPU')

Tensorflow gpu test

# As it's written in Tensorflow documentatoin: 

import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

Check gpu in tensorflow

# For tensorflow 2:
print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))

# For tensorflow 1:
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

Source: stackoverflow.com

Tensorflow gpu test

tf.test.is_gpu_available(
  cuda_only=False, min_cuda_compute_capability=None
)

Source: tensorflow.org

Similar codes

14 votes125 views1 answer
Create data dir in python
Python
3 votes112 views1 answer
Python check array exists
Python
1 vote90 views1 answer
Flask run app reset on change
Python Flask
0 vote140 views1 answer
Fernet generate key from password
Python
16 votes95 views2 answers
Python removing \n from string
Python
13 votes111 views4 answers
How to add in python
Python
0 vote114 views1 answer
Python ocr pdf dataframe
Python
11 votes98 views6 answers
Get file size python
Python
6 votes66 views2 answers
Pandas exclude columns
Python
0 vote112 views1 answer
Numpy convert 1d to 2d
Python