environment-gpu-cuda10.yml

name: mss-thesis-pytorch-gpu

channels:
  - conda-forge
  - pytorch

dependencies:
  - python=3.7
  - numpy=1.16
  - scipy=1.3
  - pytorch=1.2.0
  - cudatoolkit=10.0
  - scikit-learn=0.21
  - tqdm=4.32
  - ffmpeg
  - resampy
  - pip
  - librosa
  - pip:
    - musdb==0.3.1
    - norbert>=0.2.0
    - museval==0.3.0
    - gitpython