Pytorch使用指定GPU

1. 类似tensorflow指定GPU的方式，使用CUDA_VISIBLE_DEVICES。

1.1 直接终端中设定：

CUDA_VISIBLE_DEVICES=1 python my_script.py

1.2 python代码中设定：

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

参考
http://www.cnblogs.com/darkknightzh/p/6591923.html
2. 使用函数 set_device

import torch
torch.cuda.set_device(id)