requirements.txt

# This file may be used to create an environment using:
# $ conda create --name <env> --file <this file>
# platform: linux-64
# created-by: conda 25.3.0
_libgcc_mutex=0.1=main
_openmp_mutex=4.5=3_kmp_llvm
_sysroot_linux-64_curr_repodata_hack=3=haa98f57_10
cudatoolkit=11.8.0=h6a678d5_0
cudnn=9.8.0.87=cuda11.8
diskannpy=0.7.1=pypi_0
faiss-cpu=1.9.0=py3.10_hadc1362_0_cpu
ipdb=0.13.13=pypi_0
ipython=8.30.0=py310h06a4308_0
jupyterlab=4.2.5=py310h06a4308_0
libfaiss=1.9.0=hf65b397_0_cpu
loguru=0.7.3=pypi_0
matplotlib=3.9.2=py310h06a4308_1
networkx=3.2.1=py310h06a4308_0
notebook-shim=0.2.3=py310h06a4308_0
numpy=1.26.4=py310h5f9d8c6_0
nvitop=1.4.0=pypi_0
omegaconf=2.3.0=pypi_0
onnx=1.17.0=pypi_0
pandas=2.2.3=pypi_0
pandoc=2.12=h06a4308_3
pip=24.2=py310h06a4308_0
pyfiglet=1.0.2=pypi_0
pyg=2.6.1=py310_torch_2.4.0_cu118
python=3.10.15=he870216_1
pytorch=2.4.1=cuda118_py310h22ea9a0_300
pytorch-cuda=11.8=h7e8668a_6
pytorch-mutex=1.0=cuda
pyyaml=6.0.2=py310h5eee18b_0
scikit-learn=1.5.1=py310h1128e8f_0
scipy=1.14.1=py310h5f9d8c6_0
seaborn=0.13.2=pypi_0
setuptools=75.1.0=py310h06a4308_0
torchaudio=2.4.1=py310_cu118
torchtriton=3.0.0=py310
torchvision=0.15.2=cuda118py310h196c800_0
tqdm=4.67.1=pypi_0
wandb=0.19.6=pypi_0
wheel=0.44.0=py310h06a4308_0
yaml=0.2.5=h7b6447c_0