指定pytorch模型數據在不在cuda或者cpu上運行

# cpu
device = torch.device("cpu")

# cuda    
torch.cuda.set_device(device)

如何查看：
設置完之後，pdb打斷點輸出
print（data.device）
如果是cpu/cuda，會輸出對應信息

深入瞭解參考：https://github.com/pzhren/PyTorch/blob/4e4f81e93955090545b29ce145c1dff910f7b4b2/3%E4%B8%AD%E6%96%87%E5%AE%98%E6%96%B9%E6%95%99%E7%A8%8B/doc/1.0/notes_cuda.md