-
大小: 1.85KB文件類型: .py金幣: 1下載: 0 次發布日期: 2021-03-03
- 語言: Python
- 標簽: tensorflow??GPU??
資源簡介
【實例簡介】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
- 上一篇:pip配置阿里源
- 下一篇:SVM支持向量機(python)
評論
共有 條評論