diff options
Diffstat (limited to 'share')
3 files changed, 9 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 eecc9562a2..d244499fec 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 @@ -24,6 +24,9 @@ spack: - py-keras-preprocessing - py-keras2onnx + # MXNet + - mxnet + # PyTorch - py-botorch - py-efficientnet-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 d818feb171..4cb711402d 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 @@ -27,6 +27,9 @@ spack: - py-keras-preprocessing - py-keras2onnx + # MXNet + - mxnet + # PyTorch - py-botorch - py-efficientnet-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 728826916f..38b141c9b3 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 @@ -29,6 +29,9 @@ spack: - py-keras-preprocessing - py-keras2onnx + # MXNet + - mxnet + # PyTorch # Does not yet support Spack-install ROCm # - py-botorch |