RuntimeError: CUDA error: device kernel image is invalid

新下载的一键包，运行完install-cn.ps1后运行run_gui.ps1跳出UI界面进行的操作，新手训练模式。
训练时提示如下的日志，从网上查是版本不对or显存不够啥的，我把驱动、pytorch、CUDA都更新到最新版本了还是不行。
显存是8G专用+8G共享的，看占用率也不是很高。
这会是哪里的问题？该怎么解决？感谢各位大佬

> RuntimeError: CUDA error: device kernel image is invalid
> CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
> For debugging consider passing CUDA_LAUNCH_BLOCKING=1
> Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.
>
> 10:48:56-591986 ERROR Training failed / 训练失败