odin.yml

# conda env create -f=odin.yml
# conda activate odin
name: odin

channels:
  - conda-forge
  - rapidsai
  - nvidia
  - pytorch
  - defaults

dependencies:
  - python=3.7
  - pip>=20.3
  - cudatoolkit=11.2
  - cudnn=8.1.0
  - rapids=21.06
  - matplotlib
  - ipython
  - scikit-learn
  - pandas
  - seaborn
  - tqdm
  - dill
#  - pytorch=1.9.0
#  - torchvision
#  - torchaudio

  - pip:
      - tensorflow==2.5.0
      - tensorflow-probability==0.13.0
      - tensorflow-datasets
      - tensorflow_io
      - tensorflow-addons
      - transformers
      - kaggle
      - hydra-core>=1.0.0
      - hydra-joblib-launcher>=1.1.0
      - bigarray==0.2.1
      - numba
      - pycrypto
      - spacy
      - umap-learn
      - arviz # bayesian analysis
      - statsmodels # hypothesis testing and frequentist statistic analysis
      - typeguard>=2.10.0
#      - pyro-ppl

# Other helpful library:
# pip install git+https://github.com/DmitryUlyanov/Multicore-TSNE.git