2019年11月29日
pytorch 指定GPU训练
摘要： # 1： torch.cuda.set_device(1) # 2： device = torch.device("cuda:1") # 3：（官方推荐）import os os.environ["CUDA_VISIBLE_DEVICES"] = '1' （同时调用两块GPU的话） os.envir 阅读全文
posted @ 2019-11-29 11:21 you-wh 阅读(6497) 评论(1) 推荐(1) 编辑
Fork me on GitHub