91av视频/亚洲h视频/操亚洲美女/外国一级黄色毛片 - 国产三级三级三级三级

  • 大小: 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

評論

共有 條評論