gpuenv.yml

name: dear 
channels:
    - pytorch
    - nvidia
    - conda-forge
dependencies:
    - python==3.12.4
    - pip<24.1
    - numpy==1.26.4
    - pytorch==2.2.0
    - torchvision
    - torchaudio
    - pytorch-cuda=12.1
    - docopt
    - schema
    - cuda-cupti
    - importlib_resources
    - scikit-learn
    - pip:
        - pytorch_lightning==1.8.1
        - torch_geometric==2.5.3
        - pyg_lib
        - torch_scatter
        - -f https://data.pyg.org/whl/torch-2.2.0+cu121.html
        - wandb
        - torcheval
        - torch_sparse
        - -f https://data.pyg.org/whl/torch-2.2.0+cu121.html
        - torch_cluster
        - -f https://data.pyg.org/whl/torch-2.2.0+cu121.html
        - torch_spline_conv
        - -f https://data.pyg.org/whl/torch-2.2.0+cu121.html
        - "jax[cpu]==0.4.31"
        - -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
        - /home/dobrik/clrs/ # NOTE Change here
        - matplotlib
        - seaborn
        - git+https://github.com/HekpoMaH/torchdeq@alignment
        # - protobuf==3.20
        # - ~/clrs