python吃cpu还是显卡_tensorflow查看使用的是cpu还是gpu

方法１

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())

方法２

方法３

import tensorflow as tf

with tf.device('/gpu:0'):

a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')

b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')

c = tf.matmul(a, b)

with tf.Session() as sess:

print (sess.run(c))

安装了tensorflow-gpu,但是train的时候用的还是cpu.用方法1能检测到gpu,但实际上计算的时候还是用了cpu．用方法3可以检测出来.

import tensorflow as tf

tf.test.is_gpu_available()

找到libcudart.so所在位置,添加路径到.bashrc

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

感动,终于在gpu上跑起来了！