env.yml

name: squishy

channels:
  - conda-forge
  - defaults

dependencies:
  - cudatoolkit=10.0
  - cudnn=7.6
  - mpi4py=3.0 # installs cuda-aware openmpi
  - pip=20.0
  - python=3.7
  - tensorflow-gpu=2.0