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 03:13:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

Pool Structure

Browse Source
This commit is contained in:
weiyuqing committed 2023-03-26 19:33:03 +08:00
1 parent 6e642e87b5
commit aa569ecee1
27 files changed
+1034
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

some-test/tf_run_in_gpu.py
+1
View File
@@ -13,6 +13,7 @@ print("cpu_b:", cpu_b.device)
print("cpu_c:", cpu_c.device)
# 查看gpu是否可用
print(tf.config.list_physical_devices('GPU'))
print('asdfa')
# 指定在gpu上运行
with tf.device('/gpu:0'):
gpu_a = tf.random.normal([10000, 1000])