diff options
-rw-r--r-- | share/spack/gitlab/cloud_pipelines/stacks/ml-cpu/spack.yaml | 3 |
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 |