gp_cuda.yml

name: gp310
channels:
  - pytorch
  - nvidia
  - conda-forge
  - defaults
dependencies:
  - python=3.10
  - arviz
  - cmocean
  - jax
  - jaxlib=*=*cuda*
  - jupyterlab
  - matplotlib
  - parallel
  - pip
  - pyproj
  - python-graphviz
  - pytorch
  - pytorch-cuda=11.8
  - seaborn
  - pip:
    - ax-platform
    - hydra-core
    - hydra-zen
    - SciencePlots
    - tinygp
    - git+https://github.com/NeptuneProjects/OAOptimization.git
    - git+https://github.com/NeptuneProjects/TritonOA.git