tensorflow-gpu 버전을 설치한 후 실행환경에서 GPU를 잡고 있는지 확인할 수 있는 방법을 정리해보겠습니다.
1) tensorflow 내에서 확인
># python3
>>> from tensorflow.python.client import device_lib
>>> device_lib.list_local_devices()
...
아래와 같이 GPU가 뜨는 것을 확인할 수 있습니다.
2) nvidia-smi로 확인
terminal 창에서 nvidia-smi를 입력하면 python3 process가 GPU를 잡고 있는 것을 확인할 수 있습니다.
728x90
'개발' 카테고리의 다른 글
| 딥러닝 개발환경 세팅기(Docker, PyCharm) (1) (2) | 2021.02.09 |
|---|---|
| Docker 관련 명령어 정리 (0) | 2020.06.05 |
| Ubuntu에서 CUDA 완전 삭제 명령어 (2) | 2020.06.04 |
| Ubuntu 18.04 GPU 개발환경 세팅 (3) - CUDA 삭제 및 재설치 (0) | 2020.06.03 |
| Ubuntu 18.04 GPU 개발환경 세팅 (2) (2) | 2020.06.02 |
