# deaplearning

1개의 포스트
post-thumbnail

Tensorflow에서 GPU 사용하기

1. 개요 필자의 환경은 다음과 같다. Graphic Card: NVIDIA GeForce RTX 3060 Python: 3.8.16 Tensorflow: 2.7.0 다음과 같은 환경에서 Tensorflow를 모델을 학습하는데 있어 GPU를 사용하려 한다. 먼저 아래의 코드로 Tensorflow가 GPU를 인식하는지 알 수 있다. 출력값을 보니 Tensorflow가 GPU 인식을 못하는 것 같다! (GPU를 인식한 경우에는 다음과 같은 출력값이 포함되어 있음!) 지금부터 Tensorflow에서 GPU를 사용하기 위해 환경을 세팅해보자. 순서는 다음과 같다. NVIDIA GPU driver 설치 CUDA Toolkit 설치 cuDNN 다운로드 zlibwapi.dll (optional) 하나씩 살펴보자. 1. NVIDIA

2023년 3월 29일
·
0개의 댓글
·