Pytorch 使用常见问题查询手册

RuntimeError: CUDA error: invalid device ordinal
To use it, set CUDA_VISIBLE_DEVICES to a comma-separated list of device IDs to make only those devices visible to the application. Note that you can use this technique both to mask out devices or to change the visibility order of devices so that the CUDA runtime enumerates them in a specific order.

Error(s) in loading state_dict for DataParallel
当模型使用并行GPU加速的时候，加载参数要注意 key 的处理。 注释掉下面这一行。
常见ERROR信息：
Missing key(s) in state_dict

model_param = {k.replace('module.', ''): v for k, v in model_param.items()}