-
大小: 1.85KB文件類型: .py金幣: 1下載: 0 次發(fā)布日期: 2021-03-03
- 語言: Python
- 標(biāo)簽: tensorflow??GPU??
資源簡(jiǎn)介
【實(shí)例簡(jiǎn)介】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[‘a(chǎn)xes.unicode_minus‘]=False?
import?tensorflow?as?tf
import?timeit
cpu_data?=?[]
gpu_data?=?[]
for?n?in?range(9):
n?=?10**n
#?創(chuàng)建在CPU上運(yùn)算的2個(gè)矩陣
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)
#?創(chuàng)建使用GPU運(yùn)算的2個(gè)矩陣
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
- 上一篇:pip配置阿里源
- 下一篇:SVM支持向量機(jī)(python)
評(píng)論
共有 條評(píng)論