requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu118
accelerate==0.25.0
aiohttp==3.9.3
aiosignal==1.3.1
appdirs==1.4.4
async-timeout==4.0.3
attrs==23.2.0
av==11.0.0
black==24.1.1
certifi==2023.11.17
charset-normalizer==3.3.2
click==8.1.7
cmake==3.27.9
contourpy==1.2.0
cycler==0.12.1
docker-pycreds==0.4.0
fastrlock==0.8.2
filelock==3.13.1
fonttools==4.46.0
frozenlist==1.4.1
fsspec==2023.12.1
fvcore==0.1.5.post20221221
gitdb==4.0.11
GitPython==3.1.40
huggingface-hub==0.19.4
idna==3.6
iopath==0.1.10
iterative-stratification==0.1.7
Jinja2==3.1.2
joblib==1.3.2
kiwisolver==1.4.5
lightning==2.1.3
lightning-utilities==0.10.0
lit==17.0.6
MarkupSafe==2.1.3
matplotlib==3.8.2
mpmath==1.3.0
multidict==6.0.4
mypy-extensions==1.0.0
networkx==3.2.1
numpy==1.24.4
nvidia-cublas-cu11==11.11.3.6
nvidia-cuda-cupti-cu11==11.8.87
nvidia-cuda-nvrtc-cu11==11.8.89
nvidia-cuda-runtime-cu11==11.8.89
nvidia-cudnn-cu11==8.7.0.84
nvidia-cufft-cu11==10.9.0.58
nvidia-curand-cu11==10.3.0.86
nvidia-cusolver-cu11==11.4.1.48
nvidia-cusparse-cu11==11.7.5.86
nvidia-nccl-cu11==2.19.3
nvidia-nvtx-cu11==11.8.86
nvidia-pyindex==1.0.9
nvidia-tensorrt==99.0.0
packaging==23.2
parameterized==0.9.0
pathspec==0.12.1
Pillow==10.1.0
platformdirs==4.2.0
portalocker==2.8.2
protobuf==4.25.1
psutil==5.9.6
pyparsing==3.1.1
python-dateutil==2.8.2
pytorch-lightning==2.1.3
pytorchvideo==0.1.5
PyYAML==6.0.1
requests==2.31.0
safetensors==0.4.1
scikit-learn==1.3.2
scipy==1.11.4
sentry-sdk==1.38.0
setproctitle==1.3.3
six==1.16.0
smmap==5.0.1
spikingjelly @ git+https://github.com/fangwei123456/spikingjelly.git@671bcf4ca162a37147ffc5baa100cfb02d7df699
sympy==1.12
syops==0.0.5
tabulate==0.9.0
tensorrt==8.6.1.post1
tensorrt-bindings==8.6.1
tensorrt-libs==8.6.1
termcolor==2.4.0
threadpoolctl==3.2.0
tomli==2.0.1
torch==2.2.1+cu118
torch-tensorrt==1.4.0
torchaudio==2.2.1+cu118
torchmetrics==1.2.1
torchvision==0.17.1+cu118
tqdm==4.66.1
triton==2.2.0
typing_extensions==4.9.0
urllib3==2.1.0
wandb==0.16.1
yacs==0.1.8
yarl==1.9.4