CUDA error : device-side assert triggered

[Tony-Y]

RuntimeError: CUDA error: device-side assert triggered

I’m putting my code here: with torch.no_grad(): retrieval_one_hot = torch.zeros(k, 10).cuda() for batch_idx, (inputs, targets, indexes) in enumerate(testloader): inputs, targets, indexes = inputs.to(device), targets.to(device), indexes.to(device) net.to(device) targets = targets.cuda()#async=True) batchSize = inputs.size(0) features = net(inputs) zz = torch.zeros(batchSize * k, 10).cuda() …

Setting CUDA_LAUNCH_BLOCKING=1, you would get a better traceback.