hchang   11 Oct 2022

TPU TENSORFLOW VERSION MANAGEMENT

---

tpu vm 에서 tensorflow 버전을 바꿔주기 위해서는 libtpu.so의 버전을 같이 관리해주어야한다.

TensorFlow 버전	libtpu.so 버전
2.10.0	1.4.0
2.9.1	1.3.0
2.8.*	1.2.0
2.7.3	1.1.2

설치 방법은 위 링크로 들어가도 좋고

wget https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/tensorflow/tf-2.8.0/tensorflow-2.8.0-cp38-cp38-linux_x86_64.whl
pip3 install tensorflow-2.8.0-cp38-cp38-linux_x86_64.whl
rm tensorflow-2.8.0-cp38-cp38-linux_x86_64.whl

wget https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/libtpu/1.2.0/libtpu.so -O /lib/libtpu.so
chmod 777 /lib/libtpu.so

이런 느낌이다.