본문 바로가기
인공지능/딥러닝

tensorflow cuda 버전에 따른 설치방법

by 디찌s 2021. 3. 24.
728x90
반응형

www.tensorflow.org/install/source_windows#tested_build_configurations

 

Windows의 소스에서 빌드  |  TensorFlow

소스에서 TensorFlow pip 패키지를 빌드하고 Windows에 설치합니다. 참고: 잘 테스트되고 사전 빌드된 Windows 시스템용 TensorFlow 패키지가 이미 제공되고 있습니다. Windows용 설정 다음 빌드 도구를 설치

www.tensorflow.org

최신 gpu카드에 따라서 cudnn 설치나 cuda버전이 다르다.

 

그러므로 tensorflow도 버전을 맞춰 주어야하며, 아래에 캡처본에 따라서

 

pip install tensorflow==version

 

식으로 설치해주면된다.

 

 

 

 

728x90
반응형

'인공지능 > 딥러닝' 카테고리의 다른 글

ubuntu conda install  (0) 2020.07.20

댓글