본문 바로가기

딥러닝(Deep Learning)/컴퓨터 비전

tensorflow-gpu 패키지 설치 tip(feat. 아나콘다,cuda,cudnn)

tensorflow 패키지로 GPU와 연동을 하기 위해서는 필수적으로 CUDA 패키지와 cuDNN 패키지 설치가 필수이다. 

그러나 이게 여간 어렵고 복잡한 작업이 아니다. 

conda install tensorflow-gpu=2.5로 설치해도 관련된 패키지가 자동 설치되기는 하나, 안정적이지 않다.

위와 같이 명령하여 [cudatoolkit]를 설치하게 되면 해당 버전에 해당하는 tensorflow, tensorflow-gpu 패키지를 자동으로 설

치를 해 준다. (https://yjs-program.tistory.com/199, https://stackoverflow.com/questions/65273118/why-is-tensorflow-not-recognizing-my-gpu-after-conda-install , conda install -c conda-forge easydict 출처: https://niniit.tistory.com/28 [IT_World:티스토리]참조)

 

tensorflow 2.10 이상부터는 windows GPU 지원 x. 
https://velog.io/@hsedmr/TensorFlow-%EC%84%A4%EC%B9%98-%EC%A0%88%EC%B0%A8

cuda, cudnn 설치 요령
https://velog.io/@kimjunsu97/NVIDA-GPU-%EC%84%A4%EC%A0%95