summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--share/spack/gitlab/cloud_pipelines/stacks/ml-cpu/spack.yaml3
1 files changed, 1 insertions, 2 deletions
diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-cpu/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-cpu/spack.yaml
index a687a6928e..e6b49b9f41 100644
--- a/share/spack/gitlab/cloud_pipelines/stacks/ml-cpu/spack.yaml
+++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-cpu/spack.yaml
@@ -46,8 +46,7 @@ spack:
- py-torch
- py-torch-cluster
- py-torch-geometric
- # https://github.com/NVIDIA/apex/issues/1498
- # - py-torch-nvidia-apex
+ - py-torch-nvidia-apex
- py-torch-scatter
- py-torch-sparse
- py-torch-spline-conv