`TORCH_USE_CUDA_DSA`
时间: 2023-10-25 22:09:45 浏览: 429
pytorch中使用cuda扩展的实现示例
`TORCH_USE_CUDA_DSA` 是一个编译选项,用于启用设备端断言。当你在使用 PyTorch 进行深度学习训练时,如果出现了 `Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.` 的报错,那么可能是因为你的 PyTorch 没有启用设备端断言。要解决这个问题,你需要重新编译 PyTorch 并启用 `TORCH_USE_CUDA_DSA` 选项。不过,在大多数情况下,这个报错是由于显存不足导致的,你可以尝试降低显存或者使用 `torch.cuda.empty_cache()` 函数来释放显存。
阅读全文