반응형
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==(버전)
반응형
'딥러닝' 카테고리의 다른 글
[CNN]tensorflow 없이 numpy로만 convolution 날코딩 하기 (0) | 2020.11.10 |
---|---|
[금융]corr matrix을 이용한 계층적 클러스터링을 이용해보기 (0) | 2020.11.10 |
Classification metrics can t handle a mix of unknown and multiclass targets (0) | 2020.11.10 |
밸리데이션 로스(validation loss)가 안줄어들 경우 체크해볼 3가지 (0) | 2020.11.10 |
attention 자료 모음 (0) | 2020.11.05 |
댓글