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

[DayBeha]

问题和标题一样， 定位到是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

请问是什么问题？如何解决？

[CheKaiWei]

测试过是可以运行的，估计是cuda的问题，建议用以下语句来测试cuda能否顺利运行：
torch.cuda.is_available()