env.yml

#To install the environment, run in a terminal: conda env create -f env.yml
#Note that a version of matlab is needed for matlabengine, which is used for Fig S2. Comment out the matlabengine line if you do not have matlab installed.
name: cca
channels:
  - pytorch
  - defaults
  # - conda-forge
dependencies:
  - python=3.10
  - numpy=1.23.*
  - xarray=2022.6.*
  - notebook=6.4.*
  - pandas=1.4.*
  - scipy=1.9.*
  - matplotlib=3.5.*
  - xmltodict=0.13.*
  - jupyterlab=3.4.*
  - cython=0.29.*
  - scikit-learn=1.1.*
  - ipympl=0.9.*
  - tqdm=4.64.*
  - nodejs=18.7.*
  - pip=22.2.*
  - seaborn=0.11.*
  - cudatoolkit=11.3
  - pytorch=1.12
  - torchvision=0.13.*
  - torchaudio=0.12.*
  - pip:
    - PyPDF2==2.10
    - kornia==0.7.0
    - pytorch-metric-learning
    - matlabengine==9.13.1 #change version to match matlab version, using R2022b
    - -e packages/pyaldata #from https://github.com/NeuralAnalysis/PyalData.git
    - -e packages/config_manager #from modified from https://github.com/seblee97/config_package.git
    - -e .