簡體   English   中英

在以下CUDA錯誤“ Cuda斷言失敗:設備端斷言已觸發”之后,如何重置Colab?

[英]How to reset Colab after the following CUDA error 'Cuda assert fails: device-side assert triggered'?

我正在Google Colab上使用Pytorch運行Jupyter Notebook。 收到“ Cuda斷言失敗:設備端斷言觸發”后,我無法運行使用pytorch模塊的任何其他代碼。 有誰知道如何重置我的代碼,以便之前運行的Pytorch函數仍然可以運行?

我已經嘗試實現CUDA_LAUNCH_BLOCKING=1但是我的代碼仍然無法正常工作,因為仍然觸發了Assert!

您需要重置Colab筆記本。 要運行以前使用過的現有Pytorch模塊,您必須執行以下操作:

  • 轉到工具欄中的“運行時”
  • 點擊“重新啟動並全部運行”

這將重置您的CUDA斷言並刷新模塊,以便您可以避免出錯!

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM