前言
本人也是kaggle新手,在使用kaggle中的notebook运行本地训练好的模型作推理时,即使使用了GPU,速度依然很慢,并且可以看到GPU的占用率为0,经过多方得知在kaggle中只有最新版本的tensorflow才能使用GPU,这个版本也是kaggle镜像内置的版本,个人当然可以指定老一点的版本,直接运行
!pip install tensorflow==xxx
即可,但是这样用不了GPU,没办法,只能使用新版本的tensorflow重新训练模型了。
-------------------------------------------------------------------
更新:很奇怪,好像基于tensorflow-gpu 1.15训练的模型可以直接被2.3.1版本的tensorflow加载,GPU占用率也上去了。
参考
https://www.kaggle.com/general/135035
https://www.kaggle.com/deshmane/tensorflow-old-version
https://www.kaggle.com/product-feedback/124017