diff --git a/spack.yaml b/spack.yaml
index e91a97da39c99b840e4682c74793f41d9197081c..f00edfcc0fa2f374a82711f675b4f72c2ddfa56b 100644
--- a/spack.yaml
+++ b/spack.yaml
@@ -32,5 +32,5 @@ spack:
     # demo for codejam12
     - funniest1022 %gcc@10.3.0
     # NRP
-    #- py-torch~mkldnn~cuda~cudnn %gcc@10.3.0
+    - py-torch~mkldnn~cuda~cudnn~onnx_ml %gcc@10.3.0
   concretization: together