報錯:RuntimeError: CUDA error: all CUDA-capable devices are busy or unavailable

運行程序時報錯: 定位的錯誤是:D=torch.nn.DataParallel(discriminator(n_filters=32)).cuda(gpu_id) 本來以爲是因爲服務器上有4塊顯卡,GPU1,GPU2,GPU3的計算能力都是2.0,但是GPU0的計算能力是>3的,所以認爲是調用的時候爲每個GPU都分配一定的任務,所以才報錯的.  但是 看其他框架下的代碼直接使用XX.cuda()也

>>阅读原文<<