search

fangwei123456 / spikingjelly

SpikingJelly is an open-source deep learning framework for Spiking Neural Network (SNN) based on PyTorch.
https://spikingjelly.readthedocs.io
Other
1.22k stars 233 forks source link

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

Open DayBeha opened 1 week ago

DayBeha commented 1 week ago

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

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