fegger/trading_gnn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add pybind11 and use --no-build-isolation for PyG extensions

Browse Source 
The PyG extensions (torch-scatter, torch-sparse, etc.) need to see the
installed torch package during their build. Adding pybind11 and disabling
build isolation ensures they can locate the necessary dependencies.
This commit is contained in:
Florian Egger
2026-05-26 15:54:16 +02:00
parent a81dc690e5
commit c5459b3ee4
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Dockerfile
+3 -2
View File
@@ -62,9 +62,10 @@ RUN pip install --no-cache-dir \
ENV ROCM_HOME=/opt/rocm ENV ROCM_HOME=/opt/rocm
ENV HIP_HOME=/opt/rocm ENV HIP_HOME=/opt/rocm
RUN pip install --no-cache-dir torch-geometric==2.4.0 RUN pip install --no-cache-dir torch-geometric==2.4.0 pybind11
RUN pip install --no-cache-dir \ # Build PyG extensions with no build isolation so they can see torch
RUN pip install --no-cache-dir --no-build-isolation \
torch-scatter==2.1.2 \ torch-scatter==2.1.2 \
torch-sparse==0.6.18 \ torch-sparse==0.6.18 \
torch-cluster==1.6.2 \ torch-cluster==1.6.2 \