머신러닝 & AI 개발자 Tip & 강좌 게시판

Data Scientist와 개발자를 위한 머신러닝, AI 등 개발 Tip과 강좌 게시판 입니다. 일반적인 머신러닝은 물론 딥러닝(Tensorflow, Keras, PyTorch 등), 인공지능 관련 업무를 진행하면서 얻은 Tip이나 강좌, 새로운 소식을 적어 주시면 다른 개발자 분들에게 큰 도움이 됩니다.


GPU 머신에서 Machine Learning을 수행할때 Deep Learning 툴킷 & 패키지들이 GPU를 이용하는지 체크하는 python 코드


- GPU 기반 tensorflow나 keras를 실행할때, 속도가 느리거나 문제로 인해 체크가 필요한 경우가 있음.

- 클라우드 GPU VM이나, MLaaS 서비스를 이용할때, VM에 SSH로 접근이 어려우면 python code로 확인 해야 함.


아래의 python 코드로 코드 체크 가능


GPU 사용 여부 체크 python 코드

 

import tensorflow as tf

if tf.test.gpu_device_name(): print('Default GPU Device: {}'.format(tf.test.gpu_device_name())) else: print("Please install GPU version of TF")




Github jupyter notebook 링크


참고 링크

How to tell if tensorflow is using gpu acceleration from inside python shell? - Stack Overflow








XE Login