environment_pylops_gpu_3090.yml

name: pylops_gpu_3090
channels:
  - pytorch
  - conda-forge
  - defaults
dependencies:
  - python>=3.8
  - pip
  - numpy>=1.15.0
  - scipy>=1.4.0
  - cudatoolkit=11.3
  - scikit-learn 
  - pandas
  - pylops
  - matplotlib
  - ipython
  - jupyter
  - pytorch==1.10.1=py3.9_cuda11.3_cudnn8.2.0_0
  - torchvision
  - torchaudio
  - pytest
  - Sphinx
  - numpydoc
  - tqdm 
  - pip:
    - spgl1
    - pytorch-complex-tensor
    - cupy-cuda113
    - pytest-runner
    - setuptools_scm
    - sphinx-rtd-theme
    - sphinx-gallery
    - nbsphinx
    - image
    - torchsummary
    - torchviz