Imágenes de base de entrenamiento disponibles
ModelArts proporciona las imágenes de base impulsadas por el aprendizaje profundo, como imágenes de TensorFlow, de PyTorch y de MindSpore. En estas imágenes, se ha instalado el software obligatorio para ejecutar los trabajos de entrenamiento. Si el software de las imágenes de base no puede cumplir con los requisitos de servicio, cree nuevas imágenes basadas en las imágenes de base y utilice las nuevas imágenes para crear trabajos de entrenamiento.
Imágenes de base de entrenamiento disponibles
La siguiente tabla enumera las imágenes de ModelArts de base de entrenamiento preestablecidas.
Motor |
Versión |
---|---|
PyTorch |
pytorch_1.8.0-cuda_10.2-py_3.7-ubuntu_18.04-x86_64 |
TensorFlow |
tensorflow_2.1.0-cuda_10.1-py_3.7-ubuntu_18.04-x86_64 |
Horovod |
horovod_0.20.0-tensorflow_2.1.0-cuda_10.1-py_3.7-ubuntu_18.04-x86_64 |
horovod_0.22.1-pytorch_1.8.0-cuda_10.2-py_3.7-ubuntu_18.04-x86_64 |
|
MPI |
mindspore_1.3.0-cuda_10.1-py_3.7-ubuntu_1804-x86_64 |
Los motores de IA admitidos varían según las regiones.