2020年12月16日
pytorch设置gpu
摘要： # 获取当前设备 device = torch.device("cuda" if torch.cuda.is_available() else "cpu") 阅读全文
posted @ 2020-12-16 11:29 致林 阅读(505) 评论(0) 推荐(0) 编辑