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
'딥러닝(Deep Learning) > 컴퓨터 비전' 카테고리의 다른 글
MSE의 Weight Update의 불공정성(Feat. 분류 문제, Cross-Entropy,softmax,로그우도(Log Likelihood)) (0) | 2024.02.21 |
---|---|
yolov5을 사용하기 위한 Pytorch gpu 연동(Feat. CUDA,cudnn) (0) | 2024.01.31 |
Tensorflow - 즉시 실행 모드, 그래프 실행 모드(Earger Execution, Graph Execution)(Feat. XML 컴파일러) (0) | 2024.01.24 |
ModuleNotFoundError: No module named PyQt5(Feat. pylint) (0) | 2024.01.18 |
Snake Curve(Feat. internal Energy, image Energy) (0) | 2024.01.12 |