Pytorch 如何解决运行时错误：cuda运行时错误（59）：设备端断言触发错误

当我在colab上训练深度学习模型时，有一部分我无法理解

当我第一次训练模型时，同时测量训练损失和有效损失，它适用于每个批次和历元

但是，当我在同一单元中再次训练，或在不同单元中训练另一个模型时，

运行时错误：cuda运行时错误（59）：设备端断言触发错误

突然发生

我也在Colab窗口的最前面试过这个，但没用

import os
os.environ['CUDA_LAUNCH_BLOCKING'] = "1"

if os.environ.get('COLAB_GPU', '0') == '1':
  os.environ['GPU_NUM_DEVICES'] = '1'
  os.environ['XLA_FLAGS'] = '--xla_gpu_cuda_data_dir=/usr/local/cuda/'

我不认为这是因为损失函数的索引错误，因为 如果是索引错误，则第一次尝试不起作用！ 是我获取信息的链接，但不是我的案例

有人能解决这个问题吗