Development Study/Deep Learning

[DL] 케라스 - 텐서플로우 GPU 사용하기

octo54 2020. 11. 3. 19:15
728x90

1. CUDA 설치

- 링크 : CUDA Toolkit Arcive(https://developer.nvidia.com/cuda-toolkit-archive)

2.cuDNN 다운로드 복사 붙여 넣기

- 로그인 

- CUDA와 같은 버전 다운로드

- 링크 : developer.nvidia.com/rdp/cudnn-download

3. tensorflow-gpu 설치

- 기존에 tensorflow가 있으면 삭제 후 설치

 

728x90