RuntimeError: CUDA 错误: 遇到非法内存访问

你好，我对你的工作感到惊讶，并尝试使用你提供的代码重现它。但是，每次我尝试运行代码时，它总是无法运行，并出现了标题中提到的运行时错误。