运行代码后出现“Torch not compiled with CUDA enabled”错误
原因:电脑上没安装cuda报错
解决办法:
device = torch.device("cuda:" + str(self.params['gpu_id']) if torch.cuda.is_available() else "cpu")
把有.cuda() 的地方都换成.to(device)或者换成
.to(device="cpu")
运行代码后出现“Torch not compiled with CUDA enabled”错误
原因:电脑上没安装cuda报错
解决办法:
device = torch.device("cuda:" + str(self.params['gpu_id']) if torch.cuda.is_available() else "cpu")
把有.cuda() 的地方都换成.to(device)或者换成
.to(device="cpu")