weiyuqing/ECG_by_Diffenert_CNN_Test
1
0 Fork 0
mirror of https://github.com/This-weiyuqing/ECG_by_Diffenert_CNN_Test.git synced 2026-10-18 11:23:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

one layer CNN get

Browse Source
This commit is contained in:
weiyuqing committed 2023-03-18 13:38:39 +08:00
1 parent 14ce214397
commit 10056d33f2
742 files changed
+192 -165
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

tf_run_in_gpu.py
+48 -48
View File
@@ -2,51 +2,51 @@ import tensorflow as tf
tf.test.is_gpu_available()
# # Tensorflow 2.10是最后一个在本地windows上支持GPU的版本。从2.11版本开始，需要在windows WLS2（适用于 Linux 的 Windows 子系统）上安装才能使用GPU。所以要在native-windows上使用GPU，就只能安装2.10.0版本及以下的版本，或者安装老版的tensorflow-gpu.
# # 指定在cpu上运行
# with tf.device('/cpu:0'):
# cpu_a = tf.random.normal([10000, 1000])
# cpu_b = tf.random.normal([1000, 2000])
# cpu_c = tf.matmul(cpu_a, cpu_b)
# print("cpu_a:", cpu_a.device)
# print("cpu_b:", cpu_b.device)
# print("cpu_c:", cpu_c.device)
# # 查看gpu是否可用
# print(tf.config.list_physical_devices('GPU'))
# # 指定在gpu上运行
# with tf.device('/gpu:0'):
# gpu_a = tf.random.normal([10000, 1000])
# gpu_b = tf.random.normal([1000, 2000])
# gpu_c = tf.matmul(gpu_a, gpu_b)
# print("gpu_a:", gpu_a.device)
# print("gpu_b:", gpu_b.device)
# print("gpu_c:", gpu_c.device)
#
# import tensorflow as tf
# import timeit
#
#
# def cpu_run():
# with tf.device('/cpu:0'):
# cpu_a = tf.random.normal([10000, 1000])
# cpu_b = tf.random.normal([1000, 2000])
# c = tf.matmul(cpu_a, cpu_b)
# return c
#
#
# def gpu_run():
# with tf.device('/gpu:0'):
# gpu_a = tf.random.normal([10000, 1000])
# gpu_b = tf.random.normal([1000, 2000])
# c = tf.matmul(gpu_a, gpu_b)
# return c
#
#
# cpu_time = timeit.timeit(cpu_run, number=10)
# gpu_time = timeit.timeit(gpu_run, number=10)
# print("cpu:", cpu_time, " gpu:", gpu_time)
# # 检测 tensorflow 能使用的设备情况
# from tensorflow.python.client import device_lib
#
# # os.environ["CUDA_VISIBLE_DEVICES"] = "1" # 这个可以指定使用哪个设备
# print(device_lib.list_local_devices())
# Tensorflow 2.10是最后一个在本地windows上支持GPU的版本。从2.11版本开始，需要在windows WLS2（适用于 Linux 的 Windows 子系统）上安装才能使用GPU。所以要在native-windows上使用GPU，就只能安装2.10.0版本及以下的版本，或者安装老版的tensorflow-gpu.
# 指定在cpu上运行
with tf.device('/cpu:0'):
cpu_a = tf.random.normal([10000, 1000])
cpu_b = tf.random.normal([1000, 2000])
cpu_c = tf.matmul(cpu_a, cpu_b)
print("cpu_a:", cpu_a.device)
print("cpu_b:", cpu_b.device)
print("cpu_c:", cpu_c.device)
# 查看gpu是否可用
print(tf.config.list_physical_devices('GPU'))
# 指定在gpu上运行
with tf.device('/gpu:0'):
gpu_a = tf.random.normal([10000, 1000])
gpu_b = tf.random.normal([1000, 2000])
gpu_c = tf.matmul(gpu_a, gpu_b)
print("gpu_a:", gpu_a.device)
print("gpu_b:", gpu_b.device)
print("gpu_c:", gpu_c.device)
import tensorflow as tf
import timeit
def cpu_run():
with tf.device('/cpu:0'):
cpu_a = tf.random.normal([10000, 1000])
cpu_b = tf.random.normal([1000, 2000])
c = tf.matmul(cpu_a, cpu_b)
return c
def gpu_run():
with tf.device('/gpu:0'):
gpu_a = tf.random.normal([10000, 1000])
gpu_b = tf.random.normal([1000, 2000])
c = tf.matmul(gpu_a, gpu_b)
return c
cpu_time = timeit.timeit(cpu_run, number=10)
gpu_time = timeit.timeit(gpu_run, number=10)
print("cpu:", cpu_time, " gpu:", gpu_time)
# 检测 tensorflow 能使用的设备情况
from tensorflow.python.client import device_lib
# os.environ["CUDA_VISIBLE_DEVICES"] = "1" # 这个可以指定使用哪个设备
print(device_lib.list_local_devices())