[Python] GPU 사용 코드

GPU 확인 import tensorflow as tf from tensorflow.python.client import device_lib print(tf.__version__) print(device_lib.list_local_devices()) 2.5.0 [name: "/device:CPU:0" device_type: "CPU" memory_limit: 268435456 locality { } incarnation: 5920175265940887694 , name: "/device:GPU:0" device_type: "GPU" memory_limit: 26808418304 locality { bus_id: 1 links { } } incarnation: 14357516859692137061 physical_device_desc: "device: 0, name: Tesla V100-PCIE-32GB, pci bus id: 0000:21:00.0, compute capability...