2023年9月11日
mac的m芯片上跑cuda程序
摘要： config里 parser.add_argument('--device', type=str, default='mps') main里 device = torch.device(cfg['device']) train里 x_batch = x_batch.astype('float32') 阅读全文
posted @ 2023-09-11 18:30 辛宣 阅读(247) 评论(0) 推荐(0)