最近準(zhǔn)備參加高通的端AI的比賽，做一個(gè)目標(biāo)檢測(cè)的小應(yīng)用。涉及到目標(biāo)檢測(cè)模型的重新訓(xùn)練，從tensorflow lite的官方example里面找了一個(gè)基于pretrained model重新訓(xùn)練的教程：

https://medium.com/tensorflow/training-and-serving-a-realtime-mobile-object-detector-in-30-minutes-with-cloud-tpus-b78971cf1193

說(shuō)是30分鐘的教程，我是整整跑了一天。原因主要是tensorflow版本問(wèn)題：

- tensorflow的example都是TF1的語(yǔ)法寫的

- tensorflow的gpu安裝教程都是對(duì)TF2寫的

所以要用TF1的gpu訓(xùn)練的話，必然需要重新徹底的把nvidia driver和依賴重新裝一遍。

最終用了tensorflow-gpu==1.15， cuda=10.0的版本總算配置好了環(huán)境。