Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.
时间: 2023-11-25 12:48:50 浏览: 7687
解决AssertionError Torch not compiled with CUDA enabled.docx
`TORCH_USE_CUDA_DSA`是一个编译选项,用于启用设备端断言。如果你想启用设备端断言,可以在编译PyTorch时使用该选项。但是,如果你的问题是CUDA错误和Torch未启用CUDA的断言错误,则需要确保你的PyTorch版本已正确安装并启用了CUDA。你可以通过检查`torch.cuda.is_available()`的输出来验证CUDA是否可用。如果输出为False,则需要安装正确的CUDA版本并重新安装PyTorch。
阅读全文