environment.yml

# Notes:
# - This file only lists packages that elektronn3 *directly* depends on.
#   Transitive dependencies will be resolved by conda.
# - Dependencies are not pinned to specific versions. The minimum versions
#   listed below are the ones that are known to work. If some package causes
#   problems, just downgrade to the minimum version that is documented here.
# - If you encounter problems with CUDA versions (unwanted up-/downgrades,
#   incompatible versions), make sure to pin the cudatoolkit package to
#   a version that is known to work with your GPU driver, for example if
#   you want to stay on CUDA 8.*, run the following commands:
#       conda config --add pinned_packages 'cudatoolkit=8.*'
#       conda update --all

name: elektronn3
channels:
  - pytorch
  - conda-forge
  - defaults
dependencies:
  - colorlog >=3.1.4
  - h5py >=2.8.0
  - ipython >=6.4.0
  - matplotlib >=3.0
  - numba >=0.41.0
  - tbb
  - python >=3.6.5
  - seaborn >=0.8.1
  - tqdm >=4.23.3
  - numpy >=1.14.2
  - scikit-learn >=0.19.1
  - scikit-image >=0.14.1
  - scipy >=1.1.0
  - tensorboardx >=1.7
  - pytorch >=1.1.0  # pytorch-cpu also works, but it's not recommended.

  # Only required for running a tensorboard server:
  - tensorflow >=1.13
  - tensorboard >=1.13

  # For docs:
  - sphinx >=2.0
  - sphinx_rtd_theme >=0.4.0
  - sphinx-autodoc-typehints >=1.3.0