딥러닝(Deep Learning)/컴퓨터 비전
tensorflow-gpu 패키지 설치 tip(feat. 아나콘다,cuda,cudnn)
JIN_YOUNG _KIM
2024. 1. 26. 14:19
tensorflow 패키지로 GPU와 연동을 하기 위해서는 필수적으로 CUDA 패키지와 cuDNN 패키지 설치가 필수이다.
그러나 이게 여간 어렵고 복잡한 작업이 아니다.
위와 같이 명령하여 [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