1. 程式人生 > 其它 >AssertionError: Torch not compiled with CUDA enabled

AssertionError: Torch not compiled with CUDA enabled

技術標籤:Python3pythonerror

AssertionError: Torch not compiled with CUDA enabled
出錯位置如下:
在這裡插入圖片描述
出錯程式碼:

 index = torch.randperm(batch_size).cuda()

修改為如下即可:

 index = torch.randperm(batch_size)