trying to enable pytorch for python 3.12

CoffeaTeam · Oct 2, 2024 · efaf466 · efaf466
1 parent 4c58a3a
commit efaf466
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.9.0
   - 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