[关闭]
@songying 2020-02-18T12:52:42.000000Z 字数 224 阅读 1408

指定GPU

pytorch


1. 终端指定

  1. export CUDA_VISIBLE_DEVICES=1,2,3

2. python代码中设定

  1. import os
  2. os.environ["CUDA_VISIBLE_DEVICES"] = "1"

3. pytorch 中代码设定

  1. import torch
  2. torch.cuda.set_device(id)
  3. device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
添加新批注
在作者公开此批注前,只有你和作者可见。
回复批注