摘要： import torch #查看cuda是否可用 print(torch.cuda.is_available()) #查看cuda设备的数量 print(torch.cuda.device_count()) #查看当前使用的cuda编号 print(torch.cuda.current_device 阅读全文