You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

kg_env.yaml 4.9KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163
  1. # environment.yml ── KG-Embeddings project
  2. name: kg
  3. channels:
  4. - pytorch
  5. - defaults
  6. - nvidia
  7. - conda-forge
  8. - https://repo.anaconda.com/pkgs/main
  9. - https://repo.anaconda.com/pkgs/r
  10. dependencies:
  11. - _libgcc_mutex=0.1=conda_forge
  12. - _openmp_mutex=4.5=2_gnu
  13. - aom=3.6.1=h59595ed_0
  14. - blas=1.0=mkl
  15. - brotli-python=1.1.0=py311hfdbb021_2
  16. - bzip2=1.0.8=h4bc722e_7
  17. - ca-certificates=2024.12.14=hbcca054_0
  18. - certifi=2024.12.14=pyhd8ed1ab_0
  19. - cffi=1.17.1=py311hf29c0ef_0
  20. - charset-normalizer=3.4.1=pyhd8ed1ab_0
  21. - cpython=3.11.11=py311hd8ed1ab_1
  22. - cuda-cudart=12.4.127=0
  23. - cuda-cupti=12.4.127=0
  24. - cuda-libraries=12.4.1=0
  25. - cuda-nvrtc=12.4.127=0
  26. - cuda-nvtx=12.4.127=0
  27. - cuda-opencl=12.6.77=0
  28. - cuda-runtime=12.4.1=0
  29. - cuda-version=12.6=3
  30. - ffmpeg=4.4.2=gpl_hdf48244_113
  31. - filelock=3.16.1=pyhd8ed1ab_1
  32. - font-ttf-dejavu-sans-mono=2.37=hab24e00_0
  33. - font-ttf-inconsolata=3.000=h77eed37_0
  34. - font-ttf-source-code-pro=2.038=h77eed37_0
  35. - font-ttf-ubuntu=0.83=h77eed37_3
  36. - fontconfig=2.15.0=h7e30c49_1
  37. - fonts-conda-ecosystem=1=0
  38. - fonts-conda-forge=1=0
  39. - freetype=2.12.1=h267a509_2
  40. - gettext=0.22.5=he02047a_3
  41. - gettext-tools=0.22.5=he02047a_3
  42. - giflib=5.2.2=hd590300_0
  43. - gmp=6.3.0=hac33072_2
  44. - gmpy2=2.1.5=py311h0f6cedb_3
  45. - gnutls=3.7.9=hb077bed_0
  46. - h2=4.1.0=pyhd8ed1ab_1
  47. - hpack=4.0.0=pyhd8ed1ab_1
  48. - hyperframe=6.0.1=pyhd8ed1ab_1
  49. - idna=3.10=pyhd8ed1ab_1
  50. - intel-openmp=2022.0.1=h06a4308_3633
  51. - jinja2=3.1.5=pyhd8ed1ab_0
  52. - lame=3.100=h166bdaf_1003
  53. - lcms2=2.16=hb7c19ff_0
  54. - ld_impl_linux-64=2.43=h712a8e2_2
  55. - lerc=4.0.0=h27087fc_0
  56. - libasprintf=0.22.5=he8f35ee_3
  57. - libasprintf-devel=0.22.5=he8f35ee_3
  58. - libblas=3.9.0=16_linux64_mkl
  59. - libcblas=3.9.0=16_linux64_mkl
  60. - libcublas=12.4.5.8=0
  61. - libcufft=11.2.1.3=0
  62. - libcufile=1.11.1.6=0
  63. - libcurand=10.3.7.77=0
  64. - libcusolver=11.6.1.9=0
  65. - libcusparse=12.3.1.170=0
  66. - libdeflate=1.23=h4ddbbb0_0
  67. - libdrm=2.4.124=hb9d3cd8_0
  68. - libegl=1.7.0=ha4b6fd6_2
  69. - libexpat=2.6.4=h5888daf_0
  70. - libffi=3.4.2=h7f98852_5
  71. - libgcc=14.2.0=h77fa898_1
  72. - libgcc-ng=14.2.0=h69a702a_1
  73. - libgettextpo=0.22.5=he02047a_3
  74. - libgettextpo-devel=0.22.5=he02047a_3
  75. - libgl=1.7.0=ha4b6fd6_2
  76. - libglvnd=1.7.0=ha4b6fd6_2
  77. - libglx=1.7.0=ha4b6fd6_2
  78. - libgomp=14.2.0=h77fa898_1
  79. - libiconv=1.17=hd590300_2
  80. - libidn2=2.3.7=hd590300_0
  81. - libjpeg-turbo=3.0.0=hd590300_1
  82. - liblapack=3.9.0=16_linux64_mkl
  83. - liblzma=5.6.3=hb9d3cd8_1
  84. - libnpp=12.2.5.30=0
  85. - libnsl=2.0.1=hd590300_0
  86. - libnvfatbin=12.6.77=0
  87. - libnvjitlink=12.4.127=0
  88. - libnvjpeg=12.3.1.117=0
  89. - libpciaccess=0.18=hd590300_0
  90. - libpng=1.6.45=h943b412_0
  91. - libsqlite=3.47.2=hee588c1_0
  92. - libstdcxx=14.2.0=hc0a3c3a_1
  93. - libstdcxx-ng=14.2.0=h4852527_1
  94. - libtasn1=4.19.0=h166bdaf_0
  95. - libtiff=4.7.0=hd9ff511_3
  96. - libunistring=0.9.10=h7f98852_0
  97. - libuuid=2.38.1=h0b41bf4_0
  98. - libva=2.22.0=h8a09558_1
  99. - libvpx=1.13.1=h59595ed_0
  100. - libwebp=1.5.0=hae8dbeb_0
  101. - libwebp-base=1.5.0=h851e524_0
  102. - libxcb=1.17.0=h8a09558_0
  103. - libxcrypt=4.4.36=hd590300_1
  104. - libxml2=2.13.5=h0d44e9d_1
  105. - libzlib=1.3.1=hb9d3cd8_2
  106. - llvm-openmp=15.0.7=h0cdce71_0
  107. - markupsafe=3.0.2=py311h2dc5d0c_1
  108. - mkl=2022.1.0=hc2b9512_224
  109. - mpc=1.3.1=h24ddda3_1
  110. - mpfr=4.2.1=h90cbb55_3
  111. - mpmath=1.3.0=pyhd8ed1ab_1
  112. - ncurses=6.5=h2d0b736_2
  113. - nettle=3.9.1=h7ab15ed_0
  114. - networkx=3.4.2=pyh267e887_2
  115. - numpy=2.2.1=py311hf916aec_0
  116. - openh264=2.3.1=hcb278e6_2
  117. - openjpeg=2.5.3=h5fbd93e_0
  118. - openssl=3.4.0=h7b32b05_1
  119. - p11-kit=0.24.1=hc5aa10d_0
  120. - pillow=11.1.0=py311h1322bbf_0
  121. - pip=24.3.1=pyh8b19718_2
  122. - pthread-stubs=0.4=hb9d3cd8_1002
  123. - pycparser=2.22=pyh29332c3_1
  124. - pysocks=1.7.1=pyha55dd90_7
  125. - python=3.11.11=h9e4cc4f_1_cpython
  126. - python_abi=3.11=5_cp311
  127. - pytorch=2.5.1=py3.11_cuda12.4_cudnn9.1.0_0
  128. - pytorch-cuda=12.4=hc786d27_7
  129. - pytorch-mutex=1.0=cuda
  130. - pyyaml=6.0.2=py311h9ecbd09_1
  131. - readline=8.2=h8228510_1
  132. - requests=2.32.3=pyhd8ed1ab_1
  133. - setuptools=75.8.0=pyhff2d567_0
  134. - svt-av1=1.4.1=hcb278e6_0
  135. - sympy=1.13.3=pyh2585a3b_105
  136. - tk=8.6.13=noxft_h4845f30_101
  137. - torchaudio=2.5.1=py311_cu124
  138. - torchtriton=3.1.0=py311
  139. - torchvision=0.20.1=py311_cu124
  140. - typing_extensions=4.12.2=pyha770c72_1
  141. - tzdata=2024b=hc8b5060_0
  142. - urllib3=2.3.0=pyhd8ed1ab_0
  143. - wayland=1.23.1=h3e06ad9_0
  144. - wayland-protocols=1.37=hd8ed1ab_0
  145. - wheel=0.45.1=pyhd8ed1ab_1
  146. - x264=1!164.3095=h166bdaf_2
  147. - x265=3.5=h924138e_3
  148. - xorg-libx11=1.8.10=h4f16b4b_1
  149. - xorg-libxau=1.0.12=hb9d3cd8_0
  150. - xorg-libxdmcp=1.1.5=hb9d3cd8_0
  151. - xorg-libxext=1.3.6=hb9d3cd8_0
  152. - xorg-libxfixes=6.0.1=hb9d3cd8_0
  153. - yaml=0.2.5=h7f98852_2
  154. - zstandard=0.23.0=py311hbc35293_1
  155. - zstd=1.5.6=ha6fb4c9_0
  156. - pip:
  157. - torchmetrics>=1.4 # nicer MRR/Hits@K utilities
  158. - lovely-tensors>=0.1.0 # tensor utilities
  159. - lovely-numpy>=0.1.0 # numpy utilities
  160. - einops==0.8.1
  161. - pykeen==1.11.1
  162. - hydra-core==1.3.2
  163. - tensorboard==2.19.0