場景：
前段時間在google的colab跑faster -RCNN模型時一直都沒出問題，后面用Featurize租了一個服務器跑模型的時候，同樣的代碼卻一直報錯“CUDA error: device-side assert triggered”，
這兩天可把我逼瘋了，網上關于這種情況有好多博文介紹，大部分說的是標簽越界，有的是損失函式計算出現問題，
我只能一步步debug，最好解決了自己的問題，
'''使用GPU跑的時候，這個函式報錯“CUDA error: device-side assert triggered”'''
perm1 = torch.randperm(positive.numel(), device=positive.device)[:num_pos]
perm2 = torch.randperm(negative.numel(), device=negative.device)[:num_neg]
'''修改后，把device改為cpu'''
perm1 = torch.randperm(positive.numel(), device="cpu")[:num_pos]
perm2 = torch.randperm(negative.numel(), device="cpu")[:num_neg]
做個記錄，希望能給同樣情況的人一點幫助，
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/282399.html
標籤：其他