PyTorch時遇到了runtime error(59):device-side assert triggered at XXX debug

加入一行代碼/命令啓動腳本

1.如果使用命令來啓動腳本的話可以試試用下面這個方式來啓動腳本,理論上可以看到更加準確的報錯信息。
命令是:

CUDA_LAUNCH_BLOCKING=1

例如:

CUDA_LAUNCH_BLOCKING=1 python train.py

2.如果是Jupyter,那麼就在各種import之後加入這一行就行了

import os
os.environ[‘CUDA_LAUNCH_BLOCKING’] = “1”

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章