怎样查看是否使用 GPU 加速

2018-04-23 10:20:44.492506: I tensorflow/core/platform/cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.1 SSE4.2 AVX AVX2 FMA
2018-04-23 10:20:44.638964: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1030] Found device 0 with properties:
name: GeForce GTX 1080 Ti major: 6 minor: 1 memoryClockRate (GHz): 1.582
pciBusID: 0000:05:00.0
totalMemory: 10.91GiB freeMemory: 10.54GiB
2018-04-23 10:20:44.638992: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1120] Creating TensorFlow device (/device:GPU:0) -> (device: 0, name: GeForce GTX 1080 Ti, pci bus id: 0000:05:00.0, compute capability: 6.1)

程序运行,提示如上信息,请问,是否使用了 GPU、cuda 加速呢~ 如何设置?


提问者 tensorfyx,发表时间: 2018-4-23 11:37:58

同求,我比较笨的方法式在任务管理器里面看 GPU 性能被用了多少。


Yasin,发表于 2018-4-23 16:13:29

使用 sess = tf.Session (config=tf.ConfigProto (log_device_placement=True)) 应该可以看到张量操作所用的设备:
https://tensorflow.google.cn/programmers_guide/using_gpu#logging_device_placement


yunhai_luo,发表于 2018-4-23 16:47:21

查看 GPU 的占用情况:
ubuntu:watch -n 1 nvidia-smi


TF-boy,发表于 2018-4-24 12:51:19

楼上 +1,查看资源管理器中 GPU 的占用情况。
或者你另开一个控制台,运行一下会占用 GPU 的模型,如何这时候 freememory 比之前少了,就是之前那个程序占用了 GPU。


重庆不热,发表于 2018-7-3 16:58:20

3 楼的没试过,比较简单的方法是开启监听 nvidia-smi,自己观察


neverchange,发表于 2018-7-3 17:14:35