lightning-hydra-template/conda_env_gpu.yaml

#name: conda_env_name

channels:
  - pytorch
  - conda-forge
  - defaults

dependencies:
  - python=3.8
  - pip
  - notebook
  - cudatoolkit
  - pytorch
  - torchvision
  - torchaudio