2021年11月10日
bug-pytorch
摘要： torch 指定显卡的常见方式有一下两种 第一： import os os.environ["CUDA_VISIBLE_DEVICES"] = '1' 第二： device = torch.device('cuda:1' if torch.cuda.is_available() else 'cpu' 阅读全文
posted @ 2021-11-10 19:04 hifalee 阅读(25) 评论(0) 推荐(0) 编辑