requirements.txt

# This file may be used to create an environment using:
# $ conda create --name <env> --file <this file>
# platform: linux-64
_libgcc_mutex=0.1=main
_openmp_mutex=5.1=1_gnu
blas=1.0=mkl
bzip2=1.0.8=h7b6447c_0
ca-certificates=2023.08.22=h06a4308_0
cffi=1.16.0=pypi_0
cuda-cudart=11.8.89=0
cuda-cupti=11.8.87=0
cuda-libraries=11.8.0=0
cuda-nvrtc=11.8.89=0
cuda-nvtx=11.8.86=0
cuda-runtime=11.8.0=0
filelock=3.13.1=pypi_0
gmp=6.2.1=h295c915_3
gmpy2=2.1.2=pypi_0
intel-openmp=2023.1.0=hdb19cb5_46306
jinja2=3.1.2=pypi_0
ld_impl_linux-64=2.38=h1181459_1
libcublas=11.11.3.6=0
libcufft=10.9.0.58=0
libcufile=1.8.1.2=0
libcurand=10.3.4.101=0
libcusolver=11.4.1.48=0
libcusparse=11.7.5.86=0
libffi=3.4.4=h6a678d5_0
libgcc-ng=11.2.0=h1234567_1
libgomp=11.2.0=h1234567_1
libnpp=11.8.0.86=0
libnvjpeg=11.9.0.86=0
libstdcxx-ng=11.2.0=h1234567_1
libuuid=1.41.5=h5eee18b_0
llvm-openmp=14.0.6=h9e868ea_0
markupsafe=2.1.1=pypi_0
mkl=2023.1.0=h213fc3f_46344
mkl-fft=1.3.8=pypi_0
mkl-random=1.2.4=pypi_0
mkl-service=2.4.0=pypi_0
mkl_fft=1.3.8=py311h5eee18b_0
mkl_random=1.2.4=py311hdb19cb5_0
mpc=1.1.0=h10f8cd9_1
mpfr=4.0.2=hb69a4c5_1
mpmath=1.3.0=pypi_0
ncurses=6.4=h6a678d5_0
networkx=3.1=pypi_0
numpy=1.26.2=pypi_0
numpy-base=1.26.2=py311hf175353_0
openssl=3.0.12=h7f8727e_0
pip=23.3.1=pypi_0
pycparser=2.21=pypi_0
python=3.11.5=h955ad1f_0
pytorch=2.1.2=py3.11_cuda11.8_cudnn8.7.0_0
pytorch-cuda=11.8=h7e8668a_5
pytorch-mutex=1.0=cuda
pyyaml=6.0.1=pypi_0
readline=8.2=h5eee18b_0
setuptools=68.2.2=pypi_0
soundfile=0.12.1=pypi_0
sqlite=3.41.2=h5eee18b_0
sympy=1.12=pypi_0
tbb=2021.8.0=hdb19cb5_0
tk=8.6.12=h1ccaba5_0
torch=2.1.2=pypi_0
torchaudio=2.1.2=pypi_0
torchtriton=2.1.0=py311
tqdm=4.66.1=pypi_0
triton=2.1.0=pypi_0
typing-extensions=4.7.1=pypi_0
typing_extensions=4.7.1=py311h06a4308_0
tzdata=2023c=h04d1e81_0
wheel=0.41.2=pypi_0
xz=5.4.5=h5eee18b_0
yaml=0.2.5=h7b6447c_0
zlib=1.2.13=h5eee18b_0