click>= 8.0.3
# cudatoolkit>= 10.2
faiss-cpu
#faiss-gpu
matplotlib>= 3.5.0
pillow>= 8.4.0
pretrainedmodels>= 0.7.4
pytest>= 7.0.0
torch>= 1.10.0
scikit-image>= 0.18.3
scikit-learn>= 1.0.1
scipy>= 1.7.1
torchvision>= 0.11.1
tqdm>= 4.62.3
# torch_npu — Ascend NPU support (optional, install from official source)
# See: https://gitee.com/ascend/pytorch
# pip install torch_npu==2.1.0