feat: add requirements.txt

DKT/requirements.txt

@@ -1,5 +1,76 @@
-torch>=1.1
-torchvision
-numpy
-tqdm
-tensorboard>=1.14
+alembic==1.11.1
+appdirs==1.4.4
+certifi==2023.5.7
+charset-normalizer==3.1.0
+click==8.1.3
+cmaes==0.9.1
+cmake==3.26.3
+colorlog==6.7.0
+contourpy==1.0.7
+cycler==0.11.0
+docker-pycreds==0.4.0
+filelock==3.12.0
+fonttools==4.39.4
+fsspec==2023.5.0
+gitdb==4.0.10
+GitPython==3.1.31
+greenlet==2.0.2
+huggingface-hub==0.14.1
+idna==3.4
+importlib-metadata==6.6.0
+importlib-resources==5.12.0
+Jinja2==3.1.2
+joblib==1.2.0
+kiwisolver==1.4.4
+lightgbm==3.3.5
+lit==16.0.5
+Mako==1.2.4
+MarkupSafe==2.1.2
+matplotlib==3.7.1
+mpmath==1.3.0
+networkx==3.1
+numpy==1.24.3
+nvidia-cublas-cu11==11.10.3.66
+nvidia-cuda-cupti-cu11==11.7.101
+nvidia-cuda-nvrtc-cu11==11.7.99
+nvidia-cuda-runtime-cu11==11.7.99
+nvidia-cudnn-cu11==8.5.0.96
+nvidia-cufft-cu11==10.9.0.58
+nvidia-curand-cu11==10.2.10.91
+nvidia-cusolver-cu11==11.4.0.1
+nvidia-cusparse-cu11==11.7.4.91
+nvidia-nccl-cu11==2.14.3
+nvidia-nvtx-cu11==11.7.91
+optuna==3.2.0
+packaging==23.1
+pandas==2.0.2
+pathtools==0.1.2
+Pillow==9.5.0
+protobuf==4.23.2
+psutil==5.9.5
+pyparsing==3.0.9
+python-dateutil==2.8.2
+pytz==2023.3
+PyYAML==6.0
+regex==2023.5.5
+requests==2.31.0
+scikit-learn==1.2.2
+scipy==1.10.1
+sentry-sdk==1.24.0
+setproctitle==1.3.2
+six==1.16.0
+smmap==5.0.0
+SQLAlchemy==2.0.15
+sympy==1.12
+threadpoolctl==3.1.0
+tokenizers==0.13.3
+torch==2.0.1
+torch-geometric==2.3.1
+tqdm==4.65.0
+transformers==4.29.2
+triton==2.0.0
+typing_extensions==4.6.2
+tzdata==2023.3
+urllib3==1.26.16
+wandb==0.15.3
+zipp==3.15.0