zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update TensorFlow ecosystem (#41069)

Browse Source
This commit is contained in:
Adam J. Stewart
2024-03-11 14:33:16 +01:00
committed by GitHub
parent c72619d4db
commit 9786bd932b
7 changed files with 97 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml
View File

@@ -21,7 +21,7 @@ spack:
- py-jaxlib
# Keras
# - py-keras backend=tensorflow
- py-keras backend=tensorflow
- py-keras backend=jax
- py-keras backend=torch
- py-keras-applications

2
share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml
View File

@@ -25,7 +25,7 @@ spack:
- py-jaxlib
# Keras
# - py-keras backend=tensorflow
- py-keras backend=tensorflow
- py-keras backend=jax
- py-keras backend=torch
- py-keras-applications

2
share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml
View File

@@ -29,7 +29,7 @@ spack:
# Keras
# - py-keras backend=tensorflow
- py-keras backend=jax
# - py-keras backend=torch
- py-keras backend=torch
- py-keras-applications
- py-keras-preprocessing
- py-keras2onnx