environment.yml

name: tiktorch-server-env
channels:
  - pytorch
  - ilastik-forge
  - conda-forge
  - nodefaults
dependencies:
  # - bioimage.spec via submodule
  # - bioimage.core via submodule
  - python 3.9.*
  - numpy >=1.21,<2
  - grpcio=1.49.1 # protobuf 5 requires protoc version > 3.19.0 that requires grpcio >= 1.44
  - marshmallow-union
  - marshmallow=3.12.*
  - marshmallow-jsonschema
  - protobuf
  - pyyaml=5.3.*
  - requests
  - scikit-learn
  - scipy
  - typing-extensions
  - xarray
  - pytorch-3dunet
  - setuptools
  - pip

  # pytorch
  # remove cpuonly, add cudatoolkit and cudnn for gpu support
  - pytorch=2.3.*
  # currently it's necessary to force the cpu version, remove
  # torchvision pin when going for gpu
  # - torchvision
  - cpuonly
  # - cudatoolkit >=10.2
  # - cudnn
  - torchvision

  # tensorflow (1.14 is the latest 1.x version on cf)
  # so far we don't have any 2.x models in the model zoo
  # tensorflow skipped for now, as it conflicts with grpcio version 1.41
  # - tensorflow >=2.9,<3.0

  # bioimageio.spec / bioimageio.core dependencies:
  - annotated-types >=0.5.0,<1
  - email_validator
  - h5py
  - imageio >=2.10
  - loguru
  - packaging >=17.0
  - pooch >=1.5,<2
  - pydantic >=2.7.0,<2.10
  - pydantic-core
  - pydantic-settings >=2.5
  - python-dateutil
  - rich
  - ruyaml
  - tifffile
  - tqdm
  - typer
  - zipp


  # dev stuff
  - pytest
  - pytest-cov
  - pytest-grpc
  - pytest-timeout
  - bump2version
  - mypy
  - pre_commit

  - mkl <2024.1.0  # [linux] until pytorch is compatible with the current version