# For CUDA 11.8
pip install torch==2.3.0+cu118 -f https://download.pytorch.org/whl/torch_stable.html
# For CUDA 12.1
pip install torch==2.3.0+cu121 -f https://download.pytorch.org/whl/torch_stable.html
# For CPU only
pip install torch==2.3.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
# Latest pre-release version required
pip install spikingjelly[cuda] -U --pre