conv_fashion_mnist.py 在Cuda:0 下不能运行，设置-device cpu 才能运行

问题和标题一样， 定位到是Cuda:0 下运行到Conv2d会报段错误，直接退出程序。
```
cupy                              13.2.0
numpy                             1.22.0
spikingjelly                      0.0.0.0.14
torch                             2.3.1+cu118
torchaudio                        2.3.1+cu118
torchvision                       0.18.1+cu118
```
显卡是3060 laptop

请问是什么问题？如何解决？

