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-20 04:06:43 +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

pytorch_gpu_test.py
+9 -1
View File
@@ -1,2 +1,10 @@
import torch
torch.cuda.is_available()
torch.cuda.is_available()
print(torch.cuda.is_available())
print(torch.__version__)
print(torch.cuda.is_available() )# cuda是否可用
print(torch.cuda.device_count() )# gpu数量
print(torch.cuda.current_device())# 当前设备索引, 从0开始
print(torch.cuda.get_device_name(0))# 返回gpu名字