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”