Как сбросить Colab после следующей ошибки CUDA «Сбой Cuda: запуск на стороне устройства»?

Question

Я использую ноутбук Jupyter с помощью Pytorch в Google Colab.После того, как я получил сообщение «Cuda assert fails: сработало подтверждение на стороне устройства», я не могу запустить любой другой код, который использует мой модуль pytorch.Кто-нибудь знает, как сбросить мой код, чтобы мои функции Pytorch, которые работали до этого, все еще могли работать?

Я уже пытался реализовать CUDA_LAUNCH_BLOCKING=1, но мой код все еще не работает, так как Assert все еще запущен!

Answer 1

Вам необходимо перезагрузить ноутбук Colab.Чтобы запустить существующие модули Pytorch, которые раньше работали, вы должны сделать следующее:

• Перейти к «Runtime» на панели инструментов
• Нажмите «Перезагрузить и запустить все»

Это приведет к сбросу вашего утверждения CUDA и очистке модуля, чтобы вы могли сделать еще один удар во избежание ошибки!