celldreamer/environments/environment_gpu.yml

name: celldreamer
channels:
  - pytorch
  - nvidia
  - conda-forge
  - defaults
dependencies:
  - python=3.10
  - pytorch
  - torchvision
  - torchaudio
  - pytorch-cuda=11.8   # 12.1 for 40xx card
  - numpy<2.0          
  - pandas
  - scipy
  - scikit-learn
  - matplotlib
  - seaborn
  - scanpy
  - python-igraph
  - leidenalg
  - tqdm
  - jupyterlab
  - pip
  - tensorboard
  - pip:
    - umap-learn
    - python-box
    - yaml