텐서플로우(Tensorflow) gpu 사용 가능 여부 확인하기
2023. 8. 23. 00:22ㆍTensorflow
from tensorflow.python.client import device_lib
device_lib.list_local_devices()
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality { }
incarnation: 10360034786306285900
xla_global_id: -1,
name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 7787773952
locality { bus_id: 1 links { }
}
]
'Tensorflow' 카테고리의 다른 글
텐서플로우(Tensorflow), 케라스(Keras) 버전 확인 (0) | 2023.07.30 |
---|---|
model.load() 시 AttributeError: 'str' object has no attribute 'decode' 에러 (0) | 2023.04.17 |
텐서플로우(Tensorflow) - tensorflow==1.x 버전 for MacOS - M1 (0) | 2022.11.16 |
텐서플로우(Tensorflow) - tf.reduce_sum() 이란? (0) | 2022.11.15 |
텐서플로우(Tensorflow) - tf.gather() 란? (0) | 2022.11.15 |