在colab上跑代码,一般操作下,使用GPU需要在工具栏中选择:‘修改’->‘笔记本设置’->‘硬件加速器GPU’
最近跑的一个使用1.14版本TensorFlow的keras代码,发现训练很慢,colab提示:没有使用GPU,是否换为标准运行模式。为了解决这个问题,在stack overflow上找到了一个解决方法:
问题原因:目前colab默认TensorFlow为2.x版本,在跑我的代码之前,我使用‘!pip install tensorflow-gpu==1.14.0’重新安装了TensorFlow版本,在colab官方教程中显示
意思是说,最好不要自己用pip install安装TensorFlow,因为colab内置TensorFlow添加了一些加速器,自己安装的TensorFlow可能加速有问题。(使用pip install keras不影响GPU加速)
解决方法:要想使用1.x版本tensorflow并使用GPU,在代码运行框使用:
来将编辑器变为1.x版本。
对于这个1.x版本是什么版本的TensorFlow,以及如何指定特定版本的TensorFlow,暂不清楚。