mousert-gpu.yaml

# prereqs: build-essential (on ubuntu run sudo apt install build-essential)
# run: conda env create --file mousert-gpu.yaml
name: mouse-rt-simulate-gpu
channels:
    - conda-forge
dependencies:
    - python==3.7.13
    - requests
    - future
    - numpy
    - cython
    - ipywidgets
    - tensorflow-gpu<=1.15
    - pip
    - pip:
          - "pyhsmm @ git+https://github.com/mattjj/pyhsmm.git@master"
          - "pybasicbayes @ git+https://github.com/mattjj/pybasicbayes.git@master"
          - "autoregressive @ git+https://github.com/dattalab/pyhsmm-autoregressive.git@master"
          - "mouse-rt @ git+https://github.com/dattalab/mouse-rt@master"
          - h5py==2.10.0
          - ipykernel
          - seaborn
          - opencv-python-headless
          - "dask[complete]<=2022.04.0"
          - dask_jobqueue
          - pandas==1.3.5
          - click<8.1
          - ruamel.yaml
          - matplotlib
          - joblib
          - tqdm
          - toml
          - pyarrow
          - scipy