TensorFlow 설치 시 CUDA 관련 팁

TensorFlow를 GPU 버전으로 사용하고자 할 때 CUDA 가 필요하다. 그리고 CUDA Tool Kit을 설치할 땐 Visual studio가 설치되어 있어야 CUDA 의 components 들이 정상적으로 작동한다. 이를 위해 Visual Studio를 처음 설치하는 경우 Workload에서 C++를 이용한 데스크톱 앱 개발만을 선택하고 설치를 완료한다. 그 후에 CUDA 설치하고 TensorFlow 정상적으로 GPU 버전으로 돌아가는지 확인하면 된다.

아무것도 설치 되지 않은 상태라면,

• Python 공식 홈페이지로 가서 OS에 맞게 다운로드 한다. Windows
• 가상 환경에서 사용할 일이 많으므로 Anaconda나 Miniconda를 설치한다. Anaconda 가이드 참고, TF 가이드 Software Requirements자세하게 나와있으니 참고한다.
  • TensorFlow 설치 시 지원하는 CUDA 및 CUDA ToolKit 버전을 확인하는 것이 중요하다.
  • 그밖에 conda cheat sheet PDF 여기에 conda 사용법 요약을 확인할 수 있다.
  • Windows에서 TF 설치 방법 (블로그)