kaggle中使用低版本的tensorflow时GPU占用率很低

您所在的位置：网站首页 › gpu占用率一直为0 › kaggle中使用低版本的tensorflow时GPU占用率很低

kaggle中使用低版本的tensorflow时GPU占用率很低

2023-07-14 03:23| 来源: 网络整理| 查看: 265

前言

本人也是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

【本文地址】

kaggle中使用低版本的tensorflow时GPU占用率很低

kaggle中使用低版本的tensorflow时GPU占用率很低

今日新闻

推荐新闻