Compile with TORCH_USE_CUDA_DSA
to enable device-side assertions.
遇到这个报错,先别急着干其他的,先把device改成CPU试一试,可能就把真正的问题报出来了。
即
device='cpu'
然后重新运行代码
Compile with TORCH_USE_CUDA_DSA
to enable device-side assertions.
遇到这个报错,先别急着干其他的,先把device改成CPU试一试,可能就把真正的问题报出来了。
即
device='cpu'
然后重新运行代码