환경설정

anaconda를 사용한 tensorflow lite 환경설정

jinmc 2022. 8. 26. 10:29
반응형

tensorflow lite 에서의 anaconda 환경설정에 대해 알아보도록 하겠습니다.

우선 파이썬 버전을 3.8 이하로 맞춰주는게 중요합니다.

anaconda 환경을 만들면서, python 버전을 맞춰줍시다.

Linux 의 경우 3.5에서 3.8까지 가능하다고 합니다.

Mac의 경우 3.5에서 3.7까지 가능하다고 하네요! (주의주의!!)

참조 : 텐서플로우 라이트 환경설정

 

파이썬 3.8로 만들어봅시다.

 

conda create --name tflite python=3.8

 

conda install 이 아닌 pip로 인스톨을 해준다고 합니다.

 

python3 -m pip install tflite-runtime

 

jupyter 를 실행시키기 위해서 jupyter notebook 을 인스톨 하고, 환경을 등록하면 됩니다.

conda install jupyter notebook 등.. 자세한건 이 포스트를 참고해주세요.

 

그 후 중요한 것 중 하나는, tflite-model-maker입니다!

tflite에서 돌아가는 모델을 만들기 위해서는 tflite-model-maker를 설치해야되는데, 

그 전에는 계속 에러가 났었는데, 지금 생각해보니깐 아마도 python 버전때문에 그렇지 않을까 싶습니다.

 

pip install tflite-model-maker
반응형