딥러닝 내가 선언한 변수가 GPU에서 사용되고 있을까?
라는 의문점을 가지고 있을 수도 있습니다.
그래서 performance를 위해서 cpu에 올려야하는 변수도 있고 gpu에 있어야하는 변수가 있습니다
하지만 어디에 올라갔는지 확인 하는 방법을 가지고 왔습니다.
https://www.tensorflow.org/guide/gpu?hl=ko
GPU 사용하기 | TensorFlow Core
GPU 사용하기 컬렉션을 사용해 정리하기 내 환경설정을 기준으로 콘텐츠를 저장하고 분류하세요. TensorFlow 코드 및 tf.keras 모델은 코드를 변경할 필요 없이 단일 GPU에서 투명하게 실행됩니다. 참
www.tensorflow.org
위의 페이지에도 자세히 설명이 나왔지만,
다시 한번더 설명을 드리자면 !
tf.debugging.set_log_device_placement(True)
#위의 코드를 선언하게되면 위의 사진의 빨간색 밑 줄 처럼 어디에서 동작하는지 표시됩니다.
감사합니다.
도움이 되었다면 좋아요를...!
'Deeplearning' 카테고리의 다른 글
| nividia-smi에서 그래픽카드(graphic device) off는 무슨 의미일까? (0) | 2023.08.04 |
|---|
