search

BICLab / Spike-Driven-Transformer

Offical implementation of "Spike-driven Transformer" (NeurIPS2023)
https://openreview.net/forum?id=9FmolyOHi5
Apache License 2.0
212 stars 15 forks source link

cupy problem! #4

Closed codeinging closed 11 months ago

codeinging commented 11 months ago

I have installed cupy in conda:

image

But still encounter a problem： Traceback (most recent call last): File "train.py", line 1679, in main() File "train.py", line 976, in main TET=args.TET, File "/home/visitor/anaconda3/envs/spike_transformer_syd/lib/python3.7/site-packages/timm/models/factory.py", line 71, in create_model model = create_fn(pretrained=pretrained, pretrained_cfg=pretrained_cfg, kwargs) File "/data1/experiment_code/sunyundong/PyGT/Spike-Driven-Transformer/model/spikeformer.py", line 144, in sdt kwargs, File "/data1/experiment_code/sunyundong/PyGT/Spike-Driven-Transformer/model/spikeformer.py", line 64, in init spike_mode=spike_mode, File "/data1/experiment_code/sunyundong/PyGT/Spike-Driven-Transformer/module/sps.py", line 38, in init self.proj_lif = MultiStepLIFNode(tau=2.0, detach_reset=True, backend="cupy") File "/home/visitor/anaconda3/envs/spike_transformer_syd/lib/python3.7/site-packages/spikingjelly/clock_driven/neuron.py", line 823, in init check_backend(backend) File "/home/visitor/anaconda3/envs/spike_transformer_syd/lib/python3.7/site-packages/spikingjelly/clock_driven/neuron.py", line 30, in check_backend assert cupy is not None, 'CuPy is not installed! You can install it from "https://github.com/cupy/cupy".' AssertionError: CuPy is not installed! You can install it from "https://github.com/cupy/cupy".

jkhu29 commented 11 months ago

Thanks for your interest in this project. This issue looks more relevant to spikingjelly, please submit your issue here.

Related issues: https://github.com/zhouchenlin2096/Spikingformer/issues/1 https://github.com/fangwei123456/spikingjelly/issues/408