requirements.txt

anyio==4.4.0
astor==0.8.1
blessed==1.20.0
certifi==2024.6.2
charset-normalizer==3.3.2
decorator==5.1.1
exceptiongroup==1.2.1
filelock==3.13.1
fsspec==2024.2.0
gpustat==1.1.1
h11==0.14.0
httpcore==1.0.5
httpx==0.27.0
idna==3.7
Jinja2==3.1.3
jsmin==3.0.1
MarkupSafe==2.1.5
mpmath==1.3.0
networkx==3.2.1
numpy==1.26.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-ml-py==12.555.43
nvidia-nccl-cu11==2.20.5
nvidia-nvtx-cu11==11.8.86
opt-einsum==3.3.0
paddlepaddle-gpu==2.6.1.post112
pillow==10.3.0
protobuf==5.27.1
psutil==6.0.0
requests==2.32.3
scipy==1.13.1
six==1.16.0
sniffio==1.3.1
sympy==1.12
torch==2.3.1+cu118
torchaudio==2.3.1+cu118
torchvision==0.18.1+cu118
tqdm==4.66.4
triton==2.3.1
typing_extensions==4.12.2
urllib3==2.2.2
wcwidth==0.2.13
x2paddle==1.4.1