Imagens de base de treinamento disponíveis
O ModelArts fornece imagens de base baseadas em aprendizado profundo, como imagens de TensorFlow, PyTorch e MindSpore. Nestas imagens, o software obrigatório para a execução de trabalhos de treinamento foi instalado. Se o software nas imagens de base não puder atender aos seus requisitos de serviço, crie novas imagens com base nas imagens de base e use as novas imagens para criar trabalhos de treinamento.
Imagens de base de treinamento disponíveis
A tabela a seguir lista as imagens de base de treinamento predefinidas do ModelArts.
Mecanismo |
Versão |
---|---|
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 |
Os mecanismos de IA suportados variam dependendo das regiões.