官方镜像
更新时间:
参考本节了解平台提供的官方镜像、镜像中预置了哪些工具、软件和依赖包。
# 概述
平台为提高您的工作效率,为您提供了开箱即用的官方镜像,镜像中预置了 AI 研发常用工具、框架(TensorFlow/Pytorch)、软件包和所依赖的库,并且软件包之间的兼容性经过测试,可保证使用这些镜像构建的开发环境、训练环境是稳定可用的。
如果官方镜像中预置的依赖,不足以完成您的研发工作,您也可基于官方镜像进一步安装其他所需工具,并将其制作为您自己的镜像,以便后续使用。
# 官方镜像列表
您可以参照下表了解各官方镜像的基本信息,官方镜像中具体依赖的安装,可根据镜像名称在 平台 >镜像 模块进入镜像详情页查看 Dockerfile 了解。
其中,框架、CUDA、Python、操作系统之间需要版本兼容,官方镜像中已为您预置且经过验证,可放心使用。同时不建议您自行安装框架、CUDA、Python,以免不兼容带来非预期的错误,如需其他版本可联系趋动云官方。
名称 | 框架 | cuda | python | 操作系统 | 备注 |
---|---|---|---|---|---|
PyTorch1.6.0 Paddle2.0.2 | PyTorch1.6.0 Paddle2.0.2 | 11.0 | 3.6 | ubuntu18.04 | 框架版本过旧,不支持新库及新包。pip包较少,jupyter和tensorboard只能使用旧版本。 |
Tensorflow2.2.0 Paddle2.1.3 | Tensorflow2.2.0 Paddle2.1.3 | 11.0 | 3.6 | ubuntu18.04 | 框架版本过旧,不支持新库及新包。pip包较少,jupyter和tensorboard只能使用旧版本。 |
PyTorch2.1.0 Tensorflow2.15.0 | PyTorch2.1.0 Tensorflow2.15.0 | 12.2 | 3.10 | ubuntu22.04 | 支持分布式训练。 |
PyTorch2.0.1 Tensorflow2.13.1 Paddle2.5.2 | PyTorch2.0.1 Tensorflow2.13.1 Paddle2.5.2 | 11.8 | 3.10 | ubuntu22.04 | 支持分布式&Horovod训练。 |
PyTorch2.1.1 | PyTorch2.1.1 | 12.1 | 3.10 | ubuntu22.04 | 支持分布式训练。 |