environment.yml

name: tiktorch-server-env
channels:
  - ilastik-forge
  - pytorch
  - conda-forge
dependencies:
  # - bioimage.spec via submodule
  # - bioimage.core via submodule
  - python 3.9.*
  - numpy
  - grpcio=1.44 # 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
  - ruamel.yaml
  - scikit-learn
  - scipy
  - typing-extensions
  - xarray
  - 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
  # - tochvision

  # 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

  # convenient to use bioiamgeio.core tools
  - imageio
  - tqdm
  - typer

  # 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