name: vbeach
channels:
  - conda-forge
  - pytorch
  - nvidia
dependencies:
  - tqdm
  - pytorch
  - torchvision
  - torchaudio
  - pytorch-cuda=12.1
  - scipy
  - numpy
  - matplotlib
  - pandas
  - scikit-learn
  - opencv
  - gxx_linux-64
  - pip
  - pip:
    - torch-scatter -f https://data.pyg.org/whl/torch-2.1.2+cu121.html
    - segmentation-models-pytorch==0.2.0