environment.yml

# Use this environment file if your computer has a nvidia GPU and runs Windows or Linux.

name: sleap

channels:
    - conda-forge
    - nvidia
    - sleap
    - anaconda

dependencies:
    # Packages SLEAP uses directly
    - conda-forge::attrs >=21.2.0
    - conda-forge::cattrs ==1.1.1
    - conda-forge::imageio-ffmpeg # Required for imageio to read/write videos with ffmpeg
    - conda-forge::jsmin
    - conda-forge::jsonpickle ==1.2
    - conda-forge::networkx
    - anaconda::numpy >=1.19.5,<1.23.0
    - conda-forge::opencv <4.9.0
    - conda-forge::h5py <=3.7.0
    - conda-forge::pandas
    - conda-forge::pip
    - conda-forge::pillow #>=8.3.1,<=8.4.0
    - conda-forge::psutil
    - conda-forge::pykalman
    - conda-forge::pyside2 >=5.12  # To ensure application works correctly with QtPy.
    - conda-forge::python ~=3.7    # Run into _MAX_WINDOWS_WORKERS not found if ==
    - conda-forge::python-rapidjson
    - conda-forge::pyyaml
    - conda-forge::pyzmq
    - conda-forge::qtpy >=2.0.1
    - conda-forge::rich
    - conda-forge::scipy >=1.4.1,<=1.9.0
    - conda-forge::scikit-image
    - conda-forge::scikit-learn ==1.0
    - conda-forge::scikit-video
    - conda-forge::seaborn
    - sleap/label/dev::tensorflow ==2.7.0  # TODO: Switch to main label when updated
    - conda-forge::tensorflow-hub  # Pinned in meta.yml, but no problems here... yet
    - conda-forge::qudida
    - conda-forge::albumentations
    - conda-forge::ndx-pose <0.2.0

    # Packages required by tensorflow to find/use GPUs
    - conda-forge::cudatoolkit ==11.3.1
    # "==" results in package not found
    - conda-forge::cudnn=8.2.1
    - nvidia::cuda-nvcc=11.3

    - pip:  
        - "--editable=.[conda_dev]"