有个很好的解决办法
在程序最开始的地方加上
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
使用的时候
self.encoder = EncoderAtt(input_size=self.X.shape[1], hidden_size=encoder_hidden_size, T=T).to(device)
就是在原来
.cuda()
的地方都改成
.to(device)
这样可以自动选择是否使用cuda