在服务器上跑gpu版本tensorflow

在服务器上跑gpu版本tensorflow

  一直天真的以为学校服务器上面有gpu就可以自动的照着gpu版本的跑了，但是每次发现输出信息中的device都显示是cpu，虽然速度是比我的电脑快，但batchsize=8跑个6000张的图像一轮就要2小时，感觉不大对劲。后来一查，发现可用设备里面压根没有gpu啊？？
  首先看cuda版本的，一定要装匹配版本的gpu：借鉴了此篇

cat /usr/local/cuda/version.txt

  注意，对于服务器上有多个cuda版本要选你要的cuda的所在路径。比如，服务器上有cuda10.0、cuda9.0、cuda8.0，若选了10.0那上面的代码就是：

cat /usr/local/cuda-10.0/version.txt

  出来结果如下：

  然后再看看CUDNN 版本：（以10举例）