Youth
2022年1月16日
pytorc使用多个GPU同时训练模型
摘要： pytorch使用同一设备上多个GPU同时训练模型，只需在原有代码中稍作修改即可。 改动1： os.environ['CUDA_VISIBLE_DEVICES'] = '0,1,3' # 这里输入你的GPU_id 改动2： if torch.cuda.device_count() > 1: mode 阅读全文
posted @ 2022-01-16 22:40 lishuaics 阅读(519) 评论(0) 推荐(0) 编辑