third_party/ASpanFormer/environment.yaml

name: ASpanFormer
channels:
  - pytorch
  - conda-forge
  - defaults
dependencies:
  - python=3.8
  - cudatoolkit=10.2
  - pytorch=1.8.1
  - pip
  - pip:
      - -r requirements.txt