tensorflow-gpu 환경세팅

YulHee Kim·2021년 1월 19일
0
post-thumbnail

GPU로 딥러닝 돌리기

1. 내 컴퓨터의 GPU확인

  • 윈도우키 + X로 장치 관리자 확인
  • 디스플레이 어댑터 확인

2. NVIDIA 드라이버 다운로드

3. CUDA, cuDNN 다운로드

@@@@@@버전 주의!!

  • https://www.tensorflow.org/install/gpu
  • 텐서플로우 공식문서를 참고하는 게 답이다..

    나는 tensorflow-gpu 버전이 2.4이므로 CUDA를 v.11로 다운받아야했다.
    그리고 cuDNN은 8.0.4!
    그냥 공식문서가 답이다~~

4. 다운받은 cuDNN 파일 압축 해제 후, CUDA에 모든 파일을 각자 폴더에 맞게 복사한다.

5. 환경 변수에 들어가 설정한다 (bin, include, lib)

6. gpu 사용가능한지 확인!

끝~~ GPU를 사용하니 역시나 훨씬 빠르다 ㅠ..
지금 공부중인 CNN.. 한번 훈련할 때마다 3시간 걸리던게 1시간? 정도로 단축된 것 같다

profile
백엔드 개발자

0개의 댓글