본문 바로가기
딥러닝

Gpu 인식 못했을 경우 확인코드

by 달죽 2020. 11. 10.
반응형
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']



get_available_gpus()

 

 

 

import tensorflow as tf

print(tf.test.is_built_with_cuda())

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

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())

 

 

from keras import backend as k

k.tensorflow_backend._get_available_gpus()

 

인식 못했을때

 

관리자 모드에서 anaconda 실행

 

Pip uninstall tensorflow-gpu

Pip install tensorflow-gpu==(버전)
반응형

댓글