¿Qué marcos de IA admite ModelArts?
Los marcos y versiones de IA compatibles con ModelArts varían ligeramente según el notebook de entorno de desarrollo, los trabajos de entrenamiento y la inferencia de modelos (gestión y despliegue de aplicaciones de IA). A continuación se describen los marcos de IA soportados por cada módulo.
Notebook de entorno de desarrollo
La imagen y las versiones compatibles con las instancias de notebook de entorno de desarrollo varían según los entornos de tiempo de ejecución.
Imagen |
Descripción |
Chip soportado |
SSH remoto |
JupyterLab en línea |
---|---|---|---|---|
pytorch1.8-cuda10.2-cudnn7-ubuntu18.04 |
Imagen pública impulsada por CPU o GPU para desarrollo y entrenamiento de algoritmos generales, con el motor de IA integrado PyTorch 1.8 |
CPU o GPU |
Sí |
Sí |
mindspore1.7.0-cuda10.1-py3.7-ubuntu18.04 |
Desarrollo y entrenamiento de algoritmos generales basados en CPU o GPU, preconfigurados con el motor de IA MindSpore 1.7.0 y CUDA 10.1 |
CPU o GPU |
Sí |
Sí |
mindspore1.7.0-py3.7-ubuntu18.04 |
Desarrollo y entrenamiento de algoritmos generales con CPU, preconfigurados con el motor de IA MindSpore 1.7.0 |
CPU |
Sí |
Sí |
pytorch1.10-cuda10.2-cudnn7-ubuntu18.04 |
Desarrollo y entrenamiento de algoritmos generales basados en CPU o GPU, preconfigurados con el motor de IA PyTorch 1.10 y CUDA 10.2 |
CPU o GPU |
Sí |
Sí |
tensorflow2.1-cuda10.1-cudnn7-ubuntu18.04 |
Imagen pública impulsada por CPU o GPU para el desarrollo y entrenamiento de algoritmos generales, con el motor de IA integrado TensorFlow 2.1 |
CPU o GPU |
Sí |
Sí |
conda3-ubuntu18.04 |
Limpiar imagen base personalizada solo incluye Conda |
CPU |
Sí |
Sí |
pytorch1.4-cuda10.1-cudnn7-ubuntu18.04 |
Imagen pública impulsada por CPU o GPU para desarrollo y entrenamiento de algoritmos generales, con el motor de IA integrado PyTorch 1.4 |
CPU o GPU |
Sí |
Sí |
tensorflow1.13-cuda10.0-cudnn7-ubuntu18.04 |
Imagen pública impulsada por GPU para el desarrollo y el entrenamiento de algoritmos generales, con el motor de IA integrado TensorFlow 1.13.1 |
GPU |
Sí |
Sí |
conda3-cuda10.2-cudnn7-ubuntu18.04 |
La imagen base personalizada limpia incluye CUDA 10.2, Conda |
CPU |
Sí |
Sí |
spark2.4.5-ubuntu18.04 |
Desarrollo y entrenamiento de algoritmos con CPU, preconfigurados con PySpark 2.4.5 y se pueden conectar a clústeres de Spark preconfigurados, incluidos MRS y DLI |
CPU |
No |
Sí |
mindspore1.2.0-cuda10.1-cudnn7-ubuntu18.04 |
Imagen pública impulsada por GPU para desarrollo y entrenamiento de algoritmos, con motor de IA integrado MindSpore-GPU |
GPU |
Sí |
Sí |
mindspore1.2.0-openmpi2.1.1-ubuntu18.04 |
Imagen pública impulsada por CPU para desarrollo y entrenamiento de algoritmos, con motor de IA integrado MindSpore-CPU |
CPU |
Sí |
Sí |
Entorno de tiempo de ejecución |
Motor y versión de IA incorporados |
Chip soportado |
---|---|---|
Multi-Engine 1.0 (Python3, Recomendado) |
MXNet 1.2.1 |
CPU o GPU |
PySpark 2.3.2 |
CPU |
|
PyTorch 1.0.0 |
GPU |
|
TensorFlow 1.13.1 |
CPU o GPU |
|
TensorFlow 1.8 |
CPU o GPU |
|
XGBoost-Sklearn |
CPU |
|
Multi-Engine 1.0 (Python2) |
Caffe 1.0.0 |
CPU o GPU |
MXNet 1.2.1 |
CPU o GPU |
|
PySpark 2.3.2 |
CPU |
|
PyTorch 1.0.0 |
GPU |
|
TensorFlow 1.13.1 |
CPU o GPU |
|
TensorFlow 1.8 |
CPU o GPU |
|
XGBoost-Sklearn |
CPU |
|
Multi-Engine 2.0 (Python3) |
PyTorch 1.4.0 |
GPU |
R-3.6.1 |
CPU o GPU |
|
TensorFlow 2.1.0 |
CPU o GPU |
Trabajos de entrenamiento
En la siguiente tabla se enumeran los motores de IA.
Los motores de entrenamiento incorporados en la nueva versión se nombran en el siguiente formato:
<Training engine name_version>-[cpu | <cuda_version | cann_version >]-<py_version>-<OS name_version>-< x86_64 | aarch64>
Entorno de tiempo de ejecución |
Chip soportado |
Arquitectura del sistema |
Versión del sistema |
Motor de AI y su versión |
Versión compatible con CUDA o Ascend |
---|---|---|---|---|---|
TensorFlow |
CPU o GPU |
x86_64 |
Ubuntu 18.04 |
tensorflow_2.1.0-cuda_10.1-py_3.7-ubuntu_18.04-x86_64 |
CUDA 10.1 |
PyTorch |
CPU o GPU |
x86_64 |
Ubuntu 18.04 |
pytorch_1.8.0-cuda_10.2-py_3.7-ubuntu_18.04-x86_64 |
CUDA 10.2 |
MPI |
GPU |
x86_64 |
Ubuntu 18.04 |
mindspore_1.3.0-cuda_10.1-py_3.7-ubuntu_1804-x86_64 |
CUDA 10.1 |
Horovod |
GPU |
x86_64 |
Ubuntu 18.04 |
horovod_0.20.0-tensorflow_2.1.0-cuda_10.1-py_3.7-ubuntu_18.04-x86_64 |
CUDA 10.1 |
horovod_0.22.1-pytorch_1.8.0-cuda_10.2-py_3.7-ubuntu_18.04-x86_64 |
CUDA 10.2 |
Entorno de tiempo de ejecución |
Chip soportado |
Arquitectura del sistema |
Versión del sistema |
Motor de AI y su versión |
Versión compatible con CUDA o Ascend |
---|---|---|---|---|---|
TensorFlow |
CPU o GPU |
x86_64 |
Ubuntu 16.04 |
TF-1.8.0-python2.7 |
- |
TF-1.8.0-python3.6 |
- |
||||
TF-1.13.1-python2.7 |
- |
||||
TF-1.13.1-python3.6 |
- |
||||
TF-2.1.0-python3.6 |
- |
||||
MXNet |
CPU o GPU |
x86_64 |
Ubuntu 16.04 |
MXNet-1.2.1-python2.7 |
- |
MXNet-1.2.1-python3.6 |
- |
||||
Spark_MLlib |
CPU |
x86_64 |
Ubuntu 16.04 |
Spark-2.3.2-python3.6 |
- |
Spark-2.3.2-python2.7 |
- |
||||
Ray |
CPU o GPU |
x86_64 |
Ubuntu 16.04 |
RAY-0.7.4-python3.6 |
- |
PyTorch |
CPU o GPU |
x86_64 |
Ubuntu 16.04 |
PyTorch-1.0.0-python2.7 |
- |
PyTorch-1.0.0-python3.6 |
- |
||||
PyTorch-1.3.0-python2.7 |
- |
||||
PyTorch-1.3.0-python3.6 |
- |
||||
PyTorch-1.4.0-python3.6 |
- |
||||
Caffe |
CPU o GPU |
x86_64 |
Ubuntu 16.04 |
Caffe-1.0.0-python2.7 |
CUDA 8.0 |
MindSpore-GPU |
GPU |
x86_64 |
Ubuntu 18.04 |
MindSpore-1.1.0-python3.7 |
- |
MindSpore-1.2.0-python3.7 |
- |
Motores de IA compatibles para la inferencia de ModelArts
Si importa un modelo desde una plantilla u OBS para crear una aplicación de IA, puede seleccionar los motores y las versiones de IA en la siguiente tabla.
- Los entornos de tiempo de ejecución marcados con recommended provienen de imágenes unificadas, que se utilizarán como imágenes de inferencia base convencionales. Las imágenes unificadas proporcionan paquetes de instalación completos. Para más detalles, consulte Imágenes de base de inferencia disponibles.
- Las imágenes de la versión anterior serán descontinuadas. Utilice imágenes unificadas.
- Una imagen de tiempo de ejecución unificada se denomina con el siguiente formato: <AI engine and version> - <Hardware and version: CPU, CUDA, or CANN> - <Python version> - <SO version> - <CPU architecture>
Motor |
Entorno de ejecución |
Nota |
---|---|---|
TensorFlow |
python3.6 python2.7 tf1.13-python2.7-gpu tf1.13-python2.7-cpu tf1.13-python3.6-gpu tf1.13-python3.6-cpu tf1.13-python3.7-cpu tf1.13-python3.7-gpu tf2.1-python3.7 tensorflow_2.1.0-cuda_10.1-py_3.7-ubuntu_18.04-x86_64 (recommended) |
|
MXNet |
python3.7 python3.6 python2.7 |
|
Caffe |
python2.7 python3.6 python3.7 python2.7-gpu python3.6-gpu python3.7-gpu python2.7-cpu python3.6-cpu python3.7-cpu |
|
Spark_MLlib |
python2.7 python3.6 |
|
Scikit_Learn |
python2.7 python3.6 |
|
XGBoost |
python2.7 python3.6 |
|
PyTorch |
python2.7 python3.6 python3.7 pytorch1.4-python3.7 pytorch1.5-python3.7 pytorch_1.8.0-cuda_10.2-py_3.7-ubuntu_18.04-x86_64 (recomendado) |
|
MindSpore |
aarch64 mindspore_1.7.0-cpu-py_3.7-ubuntu_18.04-x86_64 (recommended) |
AArch64 solo puede ejecutarse en chips D310. |