智健blog

2022年6月4日
pytorch查看gpu信息，gpu是否可用
摘要： import torch gpu_use= torch.cuda.is_available() print(gpu_use) # Decide which device we want to run on device = torch.device("cuda:0" if (torch.cuda.i 阅读全文
posted @ 2022-06-04 09:45 智健 阅读(356) 评论(0) 推荐(0) 编辑