개발/Python

[Ubuntu] 텐서플로우 삭제, 다운그레이드, 버전 선택 설치

Dane.Kim 2021. 11. 15.

우분투 환경에서 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를 대신할 코드를 찾아야겠다.

댓글