Tensorflow GPU 사용이 안 되는 이유 여러가지 중 하나

Cloud_ Ghost·2024년 7월 2일
0

AI

목록 보기
33/37

설마..?

저는 window를 사용 중 입니다.
그렇다면 여기서 의아함을 느끼신 분들이 계실 겁니다.
바로 tensorflow의 version이 2.13.0이라는 것 입니다.
혹은 그게 뭐? 라고 하실수도 있습니다.
아아...
그렇습니다

참고: 기본 Windows에서의 GPU 지원은 2.10 이하 버전에서만 사용할 수 있습니다. TF 2.11부터 CUDA 빌드는 Windows에서 지원되지 않습니다. Windows에서 TensorFlow GPU를 사용하려면 WSL2에서 TensorFlow를 빌드/설치하거나 TensorFlow-DirectML-Plugin과 함께 tensorflow-cpu를 사용해야 합니다.

그래서 여태 열심히 빌드를 했는데 안 됐던건가..?하는 합리적 의심이 들기 시작했습니다..

tensorflow 제거


일단 conda uninstall tensorflow를 진행해서 tensorflow를 제거 하겠습니다.

제거 이후

제거하고 나서 conda list를 사용해서 제대로 제거가 됐는지 확인합니다.

음! 일단 제거는 확실히 됐습니다.
(하지만 저 tensorflow-gpu가 꼬롬하지만 일단 넘어가겠습니다.)

nvcc -V를 통해서 설치할 버전을 확인해보겠습니다.
cuda : 11.2, cudnn은

제거하고 나서 설명대로 tensorflow의 version을 다운그레이드해서 설치하겠습니다.

흠...근데 다시 보니 tensorflow-gpu version이 따로 있는데 굳이...?? 라는 생각이 들지만
python -veriosn을 통해서 python version이 3.9.0인걸 확인하고 그에 맞게
conda install tensorflow==2.9.0을 실행해서 설치하겠습니다.

주의!!!

음? 왜지? 하고 이유를 알아보니까 pip install tensorflow==2.9.0을 해야 설치가 된다고 합니다!

완전히 감겼습니다.
pip의 종속성 해결자가 설치된 모든 패키지의 종속성을 고려하지 않기 때문에 발생한다고 하는데 이미 설치가 되어 있는 환경에서 간단하게 해결해보려 하다가 감겨버렸습니다. 아마 패키지 충돌이나 일부 종속성이 누락된거 같은데 그냥 새롭게 콘다 가상환경을 생성하고 다시 시도해봐야 될 거 같습니다 ㅠㅠ

profile
도망쳐서 도착한 곳에 낙원이란 있을 수 없는 거야.

0개의 댓글