-
Notifications
You must be signed in to change notification settings - Fork 56
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'main' into release-1.9
- Loading branch information
Showing
9 changed files
with
78 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
2 changes: 2 additions & 0 deletions
2
build_artifacts/v2/v2.0/v2.0.0/cpu.additional_packages_env.in
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,2 @@ | ||
#Add any additional packages here | ||
conda-forge::mlflow[version='>=2.13.0,<3.0'] |
2 changes: 2 additions & 0 deletions
2
build_artifacts/v2/v2.0/v2.0.0/gpu.additional_packages_env.in
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,2 @@ | ||
#Add any additional packages here | ||
conda-forge::mlflow[version='>=2.13.2,<3.0'] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,25 @@ | ||
## Supported Image Versions | ||
|
||
The tables below list the supported SageMaker Distribution image versions. SageMaker will only release functionality and security fixes for images versions present in the tables. When you are building images or pulling images from the ECR repository, it is highly recommended for you to choose the supported image versions from the tables below. | ||
|
||
### CPU Images | ||
|
||
| Image Major Version | Image Minor Version | ECR Image URI | | ||
|---------------------|---------------------|-----------------------| | ||
|1 |1.8 |public.ecr.aws/sagemaker/sagemaker-distribution:1.8-cpu | | ||
|1 |1.7 |public.ecr.aws/sagemaker/sagemaker-distribution:1.7-cpu | | ||
|1 |1.6 |public.ecr.aws/sagemaker/sagemaker-distribution:1.6-cpu | | ||
|1 |1.5 |public.ecr.aws/sagemaker/sagemaker-distribution:1.5-cpu | | ||
|1 |1.4 |public.ecr.aws/sagemaker/sagemaker-distribution:1.4-cpu | | ||
|0 |0.12 |public.ecr.aws/sagemaker/sagemaker-distribution:0.12-cpu | | ||
|
||
### GPU Images | ||
|
||
| Image Major Version | Image Minor Version | ECR Image URI | | ||
|---------------------|---------------------|-----------------------| | ||
|1 |1.8 |public.ecr.aws/sagemaker/sagemaker-distribution:1.8-gpu | | ||
|1 |1.7 |public.ecr.aws/sagemaker/sagemaker-distribution:1.7-gpu | | ||
|1 |1.6 |public.ecr.aws/sagemaker/sagemaker-distribution:1.6-gpu | | ||
|1 |1.5 |public.ecr.aws/sagemaker/sagemaker-distribution:1.5-gpu | | ||
|1 |1.4 |public.ecr.aws/sagemaker/sagemaker-distribution:1.4-gpu | | ||
|0 |0.12 |public.ecr.aws/sagemaker/sagemaker-distribution:0.12-gpu | |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,16 @@ | ||
ARG SAGEMAKER_DISTRIBUTION_IMAGE | ||
FROM $SAGEMAKER_DISTRIBUTION_IMAGE | ||
|
||
ARG MAMBA_DOCKERFILE_ACTIVATE=1 | ||
|
||
RUN python -c "import mlflow" | ||
|
||
RUN sudo apt-get update && sudo apt-get install -y git && \ | ||
git clone --recursive https://github.com/mlflow/mlflow.git && \ | ||
: | ||
|
||
WORKDIR "mlflow/" | ||
COPY --chown=$MAMBA_USER:$MAMBA_USER scripts/run_mlflow_tests.sh . | ||
RUN chmod +x run_mlflow_tests.sh | ||
# Run tests in run_matplotlib_tests.sh | ||
CMD ["./run_mlflow_tests.sh"] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,28 @@ | ||
#!/bin/bash | ||
|
||
# Run examples for keras, pytorch, sklearn, tensorflow | ||
cd examples | ||
|
||
# keras | ||
cd keras/ | ||
python train.py | ||
cd - | ||
|
||
# pytorch | ||
cd pytorch/ | ||
python mnist_tensorboard_artifact.py | ||
cd - | ||
|
||
# sklearn | ||
for folder in "sklearn_autolog/" "sklearn_elasticnet_diabetes/linux/" "sklearn_elasticnet_wine/" "sklearn_logistic_regression/"; do | ||
cd ${folder} | ||
for file in *.py; do | ||
python "$file" || exit $? | ||
done | ||
cd - | ||
done | ||
|
||
# tensorflow | ||
cd tensorflow/ | ||
python train.py | ||
cd - |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters