我在 Google Colab 上使用 Pytorch 運行我的 Jupyter Notebook。在我收到“Cuda 斷言失?。涸O備端斷言觸發”之后，我無法運行任何其他使用我的 pytorch 模塊的代碼。有誰知道如何重置我的代碼，以便我之前工作的 Pytorch 函數仍然可以運行？我已經嘗試實現，CUDA_LAUNCH_BLOCKING=1但我的代碼仍然無法工作，因為斷言仍然被觸發！
1 回答
一只甜甜圈
TA貢獻1836條經驗 獲得超5個贊
您需要重置 Colab 筆記本。要運行以前可以工作的現有 Pytorch 模塊，您必須執行以下操作：
轉到工具欄中的“運行時”
單擊“重新啟動并運行所有”
這將重置您的 CUDA 斷言并刷新模塊，以便您可以再次嘗試避免錯誤！
添加回答
舉報
0/150
提交
取消