PyTorch 踩坑笔记

V0.3.1

RuntimeError: save_for_backward can only save input or output tensors, but argument 0 doesn't satisfy this condition

解决方案：Tensor转为Variable

input[index] 0.3.1的替代方案

torch.masked_select()

指定GPU版本
https://www.cnblogs.com/darkknightzh/p/6836568.html
1、CUDA_VISIBLE_DEVICES=1
2、import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"