PAI-DLC支持通过官方公共镜像或自定义镜像的方式进行深度学习训练。

官方公共镜像

PAI-DLC提供了多种官方公共镜像,支持不同的资源类型、Python版本及深度学习框架TensorFlow和PyTorch。官方公共镜像文件的变量列表如下。

变量名称 变量描述 变量值
region 镜像地域地址
  • cn-hangzhou:华东1(杭州)
  • cn-shanghai:华东2(上海)
  • cn-beijing:华北2(北京)
  • cn-shenzhen:华南1(深圳)
type 资源类型
  • cpu
  • gpu
  • mkl_cpu:开启了MKI计算库(MKI-DNN)
python 支持的Python版本
  • py2:支持Python 2
  • py3:支持Python 3
不同框架支持的Python版本和资源类型有差异,详细信息如下:
说明 使用镜像时,需要将${变量}替换为上表中的变量值。
  • TensorFlow 1.12(CUDA版本:nvidia-cuda-10.0-cudnn7)
    • 地域:华东1(杭州)、华东2(上海)、华北2(北京)及华南1(深圳)
    • Python版本:Python 2及Python 3
    • 资源类型:CPU、GPU及MKI_CPU
    • 镜像地址:registry.${region}.aliyuncs.com/pai-dlc/pai-tensorflow-training:1.12-${type}-${python}
    • 示例:registry.cn-beijing.aliyuncs.com/pai-dlc/pai-tensorflow-training:1.12-cpu-py2

      表示TensoFlow1.12、地域华北2(北京)、资源类型CPU及Python版本Python 2

  • TensorFlow 1.15CUDA版本:nvidia-cuda-10.0-cudnn7)
    • 地域:华东1(杭州)、华东2(上海)、华北2(北京)及华南1(深圳)
    • Python版本:Python 2及Python 3
    • 资源类型:GPU
    • 镜像地址:registry.${region}.aliyuncs.com/pai-dlc/pai-tensorflow-training:1.15-${type}-${python}
    • 示例:registry.cn-beijing.aliyuncs.com/pai-dlc/pai-tensorflow-training:1.15-gpu-py2

      表示TensoFlow1.15、地域华北2(北京)、资源类型GPU及Python版本Python 2

  • TensorFlow 2.2(CUDA版本:nvidia-cuda-10.1-cudnn7)
    • 地域:华东1(杭州)、华东2(上海)、华北2(北京)及华南1(深圳)
    • Python版本:Python 3
    • 资源类型:CPU及GPU
    • 镜像地址:registry.${region}.aliyuncs.com/pai-dlc/pai-tensorflow-training:2.2-${type}-${python}
    • 示例:registry.cn-beijing.aliyuncs.com/pai-dlc/pai-tensorflow-training:2.2-cpu-py3

      表示TensoFlow2.2、地域华北2(北京)、资源类型CPU及Python版本Python 3

  • PyTorch 1.3(CUDA版本:nvidia-cuda-10.0-cudnn)
    • 地域:华东1(杭州)、华东2(上海)、华北2(北京)及华南1(深圳)
    • Python版本:Python 2及Python 3
    • 支持资源类型:CPU、GPU及MKI_CPU
    • 镜像地址:registry.${region}.aliyuncs.com/pai-dlc/pai-pytorch-training:1.3-${type}-${python}
    • 示例:registry.cn-beijing.aliyuncs.com/pai-dlc/pai-pytorch-training:1.3-gpu-py3

      表示PyTorch1.3、地域华北2(北京)、资源类型GPU及Python版本Python 3

  • PyTorch1.5(CUDA版本:nvidia-cuda-10.1-cudnn7)
    • 地域:华东1(杭州)、华东2(上海)、华北2(北京)及华南1(深圳)
    • Python版本:Python 2及Python 3
    • 资源类型:CPU、GPU及MKI_CPU
    • 镜像地址:registry.${region}.aliyuncs.com/pai-dlc/pai-pytorch-training:1.5-${type}-${python}
    • 示例:registry.cn-beijing.aliyuncs.com/pai-dlc/pai-pytorch-training:1.5-gpu-py3

      表示PyTorch1.5、地域华北2(北京)、资源类型GPU及Python版本Python 3

自定义镜像

PAI-DLC支持自定义镜像,您可以根据自己的业务场景,制作镜像文件,详情请参见Docker 镜像基本操作。PAI-DLC提供的基本镜像文件为:reg.docker.alibaba-inc.com/dlc/mirror:nvidia-cuda-10.0-cudnn7-devel-ubuntu16.04