-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathenvironment.yml
More file actions
36 lines (36 loc) · 911 Bytes
/
environment.yml
File metadata and controls
36 lines (36 loc) · 911 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
# Conda environment for dualgnn_paper.
# Install: conda env create -f environment.yml
#
# Default torch is the cu130 CUDA build (includes Blackwell / sm_120).
# For a CPU-only torch, replace the cu130 index-url below with the CPU
# wheel index: https://download.pytorch.org/whl/cpu
name: dualgnn
channels:
- conda-forge
dependencies:
- python>=3.10
- cython
- numba
- numpy
- scipy
- polars
- matplotlib
- jupyterlab
- pytest
- setuptools
# native libs for CYTools (polytope arithmetic, lattice enumeration)
- ppl
- python-flint
- normaliz
- pip
- pip:
- pplpy
- pypalp
- regfans
- git+https://github.com/LiamMcAllisterGroup/cytools.git
- tensorboard
# torch with Blackwell-supporting cu130 wheels (see top for CPU alt)
- --index-url https://download.pytorch.org/whl/cu130
- --extra-index-url https://pypi.org/simple
- torch>=2.7
- -e .