tensorflow测试gpu代码_tensorflow-gpu1.15.0测试和调用

1.TensorFlow-gpu测试

首先进入python，在python下，依次使用以下指令。观察输出的参数是否是gpu版本即可

import tensorflow as tfsess = tf.Session()     #这句运行后会返回GPU的信息，如图3.1所示。若只有cpu信息，则安装失败a = tf.constant([1,2,3],shape=[3],name='a')b = tf.constant([1,2,3],shape=[3],name='b')c=a+b12345

图1.1 sess=tf.Session返回gpu信息效果图

图1.1

这里使用的四块rtx2080 ti gpu,信息均已返回
至此安装结束。

4.TensorFlow-gpu实际调用设置

在实际项目中，在调用TensorFlow-gpu时，还需要进行设置，设置代码如下，只需要将下列代码放在python文件中即可。1

import tensorflow as tfimport osos.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"   os.environ["CUDA_VISIBLE_DEVICES"] = "0"    #这里是gpu的序号，指定使用的gpu对象config = tf.ConfigProto()config.gpu_options.allow_growth = True