Tensorflow-gpu运行代码慢的问题排查
总结一次踩坑经历。
1. 查看代码是否运行在Tensorflow-gpu版本
检查是否同时安装了Tensorflow和Tensorflow-gpu两个版本,如果都安装了,卸载cpu版本的。
2. 查看GPU占用情况
一边运行代码,一边查看此时GPU的占用情况,可以反映出这个代码是否用了GPU加速。Ctrl+R进入命令窗口,在C:\Program Files\NVIDIA Corporation\NVSMI 路径下输入命令 nvidia-smi 查看GPU的占用情况。
这个图说明GPU占用了80%,说明代码正在使用GPU,也可以终止代码运行再查看一次。