diff options
author | Adam J. Stewart <ajstewart426@gmail.com> | 2023-03-17 03:13:44 -0600 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-03-17 10:13:44 +0100 |
commit | a84fb716a0c2303b15d95e2dc016a8c1d2029fc8 (patch) | |
tree | b7f3c7b6124d8f24f1fdb342f98346cf42b1545f /share/spack | |
parent | a11f06885f2c6b8d6c8a09e847aa99c31b742f0f (diff) | |
download | spack-a84fb716a0c2303b15d95e2dc016a8c1d2029fc8.tar.gz spack-a84fb716a0c2303b15d95e2dc016a8c1d2029fc8.tar.bz2 spack-a84fb716a0c2303b15d95e2dc016a8c1d2029fc8.tar.xz spack-a84fb716a0c2303b15d95e2dc016a8c1d2029fc8.zip |
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
Diffstat (limited to 'share/spack')
3 files changed, 3 insertions, 0 deletions
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 |