우분투 환경에서 pipenv shell 안에서,
pip uninstall tensorflow
pip uninstall tensorflow-gpu
pip uninstall tensorflow-cpu
로 설치된 텐서플로를 모두 지워주고,
pip install tensorflow==2.5.0 처럼 원하는 버전을 선택해 설치할 수 있다.
gpu, cpu도 마찬가지 방법으로 가능하다.
2.6.0 버전 이후로는 predict_classes가 없어져서, 일단 임시방편으로 버전 다운그레이드를 선택했다.
나중에 문제가 있다면 다시 버전을 올리고, predict_classes를 대신할 코드를 찾아야겠다.
댓글