trying to enable pytorch for python 3.12

CoffeaTeam · Sep 11, 2024 · 3c1a205 · 3c1a205
1 parent 1abc9aa
commit 3c1a205
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/coffea-dask/environment.yaml b/coffea-dask/environment.yaml
@@ -53,13 +53,15 @@ dependencies:
   - coffea=2024.8.1
   - rucio-clients
     # pyg
-  - pyg
+  #- pyg
+  - pytorch
+  - cpuonly
   - pytorch-cluster
   - pytorch-sparse
   - pytorch-spline-conv
   - pip:
     - fastjet # to be added to conda-forge: https://github.com/scikit-hep/fastjet/issues/133
     - tritonclient[all]
-    - tflite-runtime==2.14.0
+    #- tflite-runtime # not available for Python 3.12
     - onnxruntime
     - fsspec-xrootd