torchbenchmark/models/demucs/environment-cuda.yml

name: demucs

channels:
  - conda-forge
  - pytorch

dependencies:
  - python=3.7
  - cudatoolkit=10
  - ffmpeg==4.2
  - pytorch=1.4.0
  - scipy==1.3.1
  - tqdm>=4.36.1
  - pip
  - pip:
    - lameenc==1.2.2
    - musdb==0.3.1
    - museval==0.3.0
    - requests==2.22
    - treetable==0.2.3