Computer Science/Linux
GPU와 쿠다가 연동되고 있는지 확인 하는 방법
버터젤리
2020. 1. 17. 17:45
텐서플로우(tensorflow) GPU버전 사용하기 · jaehyeong's ds
jaehyeongan.github.io
GPU와 쿠다가 연동되고 있는지 확인 하는 방법
tensorflow가 GPU버전으로 잘 설치되었고, 나의 GPU를 잘 인식하고 있는지 확인하고 싶다면 아래와 같은 코드를 통해 확인할 수 있다. tensorflow가 인식하는 로컬 device 목록을 보여주게 된다.
$ from tensorflow.pyhton.client import device_lib
$ device_lib.list_local_devices()