environment_cuda117.yaml

name: spanet

channels:
  - pytorch
  - nvidia
  - conda-forge
  - nodefaults

dependencies:
  - python=3.10.*
  - conda
  - pip
  - onnx

  - numpy=1.24
  - sympy
  - scikit-learn
  - numba
  - opt_einsum
  - h5py
  - cytoolz

  - pytorch=2.0.1
  - pytorch-lightning=2.0.7
  - pytorch-cuda=11.7
  - torchvision
  - torchaudio

  - tensorboard
  - tensorboardx

  - jupyterlab
  - seaborn
  - rich