起因
上个月做卷积神经网络的手势识别,一开始用CPU对模型进行训练,直接练了我二十几个小时,还因为电脑电源有问题,在最后关头失败了。最后只能减少训练次数来完成模型训练。
经过相关文章查阅,发现实际上只要使用GPU来训练模型,效率就有极大的提升。但是TensorFlow的安装又繁琐又有各种版本限制版本匹配之类的问题,也是搞了两天才把TensorFlow的GPU环境弄上。
前两天电脑又太老旧,硬盘直接报废,换了个新硬盘,环境全都得重配,当时配环境的时候参考了那么多文章就没自己写文章记录了。现在换完硬盘后重配GPU环境,再去搜那些文章,零零散散,没有集中记录,太难查阅。因而在自己这次成功配回环境后,记录一下自己这次重配的过程。
查看TensorFlow环境是CPU还是GPU
通过这个代码语句可以获知系统当前TensorFlow环境
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices(