mirror of
https://github.com/This-weiyuqing/ECG_by_Diffenert_CNN_Test.git
synced 2026-10-18 03:13:39 +08:00
11 lines
317 B
Python
11 lines
317 B
Python
import torch
|
|
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名字
|