Pytorch错误：cuda runtime error (59): device-side assert triggered at THC

参考资料：https://blog.csdn.net/Geek_of_CSDN/article/details/86527107
  今天在复现一份代码的过程中，遇到了一个很神奇的错误——RuntimeError: cuda runtime error (59): device-side assert triggered at …/aten/src/THC/THCCachingHostAllocator.cpp:265，如下图所示。

错误提示

  看到这个错误，第一反应是很懵的，因为从错误提示上看不到任何出错来源。在网上搜到的大部分资料都是建议检查一下label是不是从0开始。由于这份代码读取数据的方式是Pytorch自带的’torchvision.datasets.ImageFolder’和’torch.utils.data.DataLoader’函数，输出的label应该是没问题的，检查之后也是如此（可以使用assert断言进行检查）。继续查找资料，发现有人说可以在代码中加入如下部分进一步定位出错位置：

import os
os.environ[