environment_pylops_gpu.yml

name: pylops_gpu
channels:
  - pytorch
  - conda-forge
  - defaults
dependencies:
  - python>=3.6.4
  - pip
  - numpy>=1.15.0
  - scipy>=1.4.0
  - cudatoolkit=10.2 
  - scikit-learn 
  - pandas
  - pylops
  - matplotlib
  - ipython
  - jupyter
  - pytorch
  - torchvision
  - torchaudio
  - pytest
  - Sphinx
  - numpydoc
  - tqdm 
  - pip:
    - spgl1
    - pytorch-complex-tensor
    - cupy-cuda102
    - pytest-runner
    - setuptools_scm
    - sphinx-rtd-theme
    - sphinx-gallery
    - nbsphinx
    - image
    - torchsummary
    - torchviz