From a84fb716a0c2303b15d95e2dc016a8c1d2029fc8 Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Fri, 17 Mar 2023 03:13:44 -0600 Subject: Update the PyTorch ecosystem (#36132) * py-pytorch-lightning: add v2.0.0 * py-lightning-utilities: add v0.8.0 * Update all PyTorch packages * Open-CE does not yet have patches for PyTorch 2 on ppc64le --- share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml | 1 + .../spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml | 1 + .../spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml | 1 + 3 files changed, 3 insertions(+) (limited to 'share') diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml index 0e4a9e7d90..c9ca7f5b34 100644 --- a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml +++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml @@ -28,6 +28,7 @@ spack: - py-efficientnet-pytorch - py-gpytorch - py-kornia + - py-lightning - py-pytorch-gradual-warmup-lr - py-pytorch-lightning - py-segmentation-models-pytorch diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml index 6b0501b9bb..f4876c6bc5 100644 --- a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml +++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml @@ -31,6 +31,7 @@ spack: - py-efficientnet-pytorch - py-gpytorch - py-kornia + - py-lightning - py-pytorch-gradual-warmup-lr - py-pytorch-lightning - py-segmentation-models-pytorch diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml index 26946d9025..956cbf3ff8 100644 --- a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml +++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml @@ -34,6 +34,7 @@ spack: # - py-efficientnet-pytorch # - py-gpytorch # - py-kornia + # - py-lightning # - py-pytorch-gradual-warmup-lr # - py-pytorch-lightning # - py-segmentation-models-pytorch -- cgit v1.2.3-60-g2f50