environment.yml

name: basic
channels:
  - nvidia
  - pytorch
  - defaults
  - conda-forge
dependencies:
  - python=3.9.17
  - pip=23.3.1
  - pytorch=1.13.1
  - pip:
    - -r requirements.txt