env.yaml

name: la
channels:
  - defaults
  - pytorch
  - nvidia

dependencies:
  - python=3.9
  - pytorch==1.13.0
  - pytorch-cuda=11.6
  - torchvision
  - torchaudio
  - pip
  - pip:
    - -e .[dev]