더북(TheBook)

2. CUDA Toolkit을 OS 버전에 맞게 설치

PC의 OS 버전과 비트를 확인한 후 해당 CUDA를 설치한다. CUDA는 일반적으로 C:\ Program Files\NVIDIA GPU Computing Toolkit 폴더에 설치된다.

URL https://developer.nvidia.com/cuda-toolkit-archive

이때 CUDA, cuDNN과 텐서플로의 버전이 호환되는 것이 중요하다. 다음 주소에서 호환 여부를 확인하자.

URL https://www.tensorflow.org/install/source#tested_build_configurations

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.