check if tensorflow gpu is installed
import tensorflow as tf
print(tf.test.gpu_device_name())
check if tensorflow gpu is installed
import tensorflow as tf
print(tf.test.gpu_device_name())
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')))
tensorflow gpu test
tf.test.is_gpu_available(
cuda_only=False, min_cuda_compute_capability=None
)
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))
python check my gpu
from tensorflow.python.client import device_lib
def get_available_gpus():
local_device_protos = device_lib.list_local_devices()
return [x.name for x in local_device_protos if x.device_type == 'GPU']
Copyright © 2021 Codeinu
Forgot your account's password or having trouble logging into your Account? Don't worry, we'll help you to get back your account. Enter your email address and we'll send you a recovery link to reset your password. If you are experiencing problems resetting your password contact us