Docker tensorflow 2.16.1 설치, 환경 설정

python version = 3.9.19
latest tensorflow version = 2.16.1

설치시 hd5f관련 에러가 계속 떴다. 이를 해결하기 위해 hd5f를 설치하고자 했으나 pip install hd5f도 되지 않는다. 사실 hd5f는 tensorflow설치시 자동으로 설치되므로 hd5f설치가 안되는 것을 해결해야한다.

컨테이너에 접속해서 libhdf5-dev를 우선 설치하고
run apt-get update && apt-get install -y libhdf5-dev를 설치해준다.(pip install 아님.)

그러고 tensorflow 공식 홈페이지로부터 버전에 맞게 pip install 한다.

pip install https://storage.googleapis.com/tensorflow/versions/2.16.1/tensorflow-2.16.1-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

tensorflow 2.16의 issue사항을 확인해보니

tf.keras를 사용하기 위해서는 tf-keras~=2.16을 설치해야하고, tf.estimator은 삭제됬다고 한다.

tf.keras.models.load_model이 필요하므로 tf-keras를 설치해주었다.

https://blog.tensorflow.org/2024/03/whats-new-in-tensorflow-216.html