AssertionError: Torch not compiled with CUDA enabled
出错位置如下:
出错代码:
index = torch.randperm(batch_size).cuda()
修改为如下即可:
index = torch.randperm(batch_size)
AssertionError: Torch not compiled with CUDA enabled
出错位置如下:
出错代码:
index = torch.randperm(batch_size).cuda()
修改为如下即可:
index = torch.randperm(batch_size)