Tensorflow gpu_accelerate

【核心代码】Tensorflow gpu_accelerate

import  numpy as np
import  matplotlib
from    matplotlib import pyplot as plt
# Default parameters for plots
matplotlib.rcParams[‘font.size‘] = 20
matplotlib.rcParams[‘figure.titlesize‘] = 20
matplotlib.rcParams[‘figure.figsize‘] = [9 7]
matplotlib.rcParams[‘font.family‘] = [‘STKaiti‘]
matplotlib.rcParams[‘axes.unicode_minus‘]=False 



import tensorflow as tf
import timeit




cpu_data = []
gpu_data = []
for n in range（9）:
	n = 10**n
	# 创建在CPU上运算的2个矩阵
	with tf.device（‘/cpu:0‘）:
		cpu_a = tf.random.normal（[1 n]）
		cpu_b = tf.random.normal（[n 1]）
		print（cpu_a.device cpu_b.device）
	# 创建使用GPU运算的2个矩阵
	with tf.device（‘/gpu:0‘）:
		gpu_a = tf.random.normal（[1 n]）
		gpu_b = tf.random.normal（[n 1]）
		print（gpu_a.device gpu_b.device）

	def cpu_run（）:
		with tf.device（‘/cpu:0‘）:
			c = tf.matmul（cpu_a cpu_b）
		return c 

	def gpu_run（）:
		with tf.device（‘/gpu:0‘）:
			c = t