pytorch 几个基本问题

1. 如何 finetune一个已有模型

参考 PyTorch参数初始化和Finetune
利用requires_grad参数，将需要更新参数的层设置为True即可，示例：

from torch import modelsmodel = models.vgg19_bn(pretrained=True)for param in model.parameters():    param.requires_grad = False

2. 如何更改GPU的卡号

参考PyTorch中使用指定的GPU,
使用CUDA_VISIBLE_DEVICES参数，示例如下：

import osos.environ["CUDA_VISIBLE_DEVICES"] = "1"

3. 如何使用指定的多 GPU

参考pytorch-multi-gpu,
示例：

model = nn.DataParallel(model.cuda(1), device_ids=[1,2,3,4,5])

4. 如何自动使用空闲的 GPU

参考如何为TensorFlow和PyTorch自动选择空闲GPU，解决抢卡争端