Pytorch运行“Torch not compiled with CUDA enabled”

遇到了没有预训练模型的问题

model.load_state_dict(torch.load('pretrain_model_D2/D2_DenseNet121', map_location='cpu'), strict=False)

#掉上面那句话之后出现了

AssertionError: Torch not compiled with CUDA enabled

解决办法：Pytorch使用CPU运行“Torch not compiled with CUDA enabled”_ssjdoudou的博客-CSDN博客

在程序最开始的地方加上

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降级之后就好了