利用GPU訓練時的常見錯誤

1.CUDA_VIDIBLE_DEVICES=4,5 python3 main.py   我想在集羣條件下利用4,5號GPU，因爲模型較小，並不清楚是均可以指定4,5號GPU（內存佔用狀況看不出來。。。後續會找找其餘驗證辦法），報了以下幾個錯誤：python all tensors must be on devices[0]：緩存 CUDNN_STATUS_INTERNAL_ERROR：目前找到了

>>阅读原文<<