In order to use GPU version tensorflow, remember to delete CPU version tf if exists.
If you are using GPU version tf, when you build a session, GPU info will be printed.
如果两个一起装了,直接import tf的时候,会默认使用非gpu版本……结果显存占用为0
卸载了cpu版本的tensorflow之后,再使用Import tf就会使用gpu版本.
查看是否使用gpu很简单,在建立session的时候会有提示信息的。下图就是正确使用gpu版本的时候的提示信息。