Help Center/ ModelArts/ Image Management/ Using a Preset Image/ Inference Base Images/ PyTorch (CPU/GPU)-powered Inference Base Images
Updated on 2023-04-17 GMT+08:00

PyTorch (CPU/GPU)-powered Inference Base Images

ModelArts provides the following inference base images powered by PyTorch (CPU/GPU):

Engine Version 1: pytorch_1.8.0-cuda_10.2-py_3.7-ubuntu_18.04-x86_64

  • Image path: swr.{region_id}.myhuaweicloud.com/atelier/pytorch_1_8:pytorch_1.8.0-cuda_10.2-py_3.7-ubuntu_18.04-x86_64-20221118143845-d65d817
  • Image creation time: 20220713110657 (yyyy-mm-dd-hh-mm-ss)
  • Image system version: Ubuntu 18.04.4 LTS
  • CUDA: 10.2.89
  • cuDNN: 7.6.5.32
  • Path and version of the Python interpreter: /home/ma-user/anaconda3/envs/PyTorch-1.8/bin/python, python 3.7.10
  • Third-party package installation path: /home/ma-user/anaconda3/envs/PyTorch-1.8/lib/python3.7/site-packages
  • Certain pip installation packages:
    Cython                         0.27.3
    easydict                       1.9
    Flask                          2.0.1
    fonttools                      4.34.4
    gunicorn                       20.1.0
    ipykernel                      6.7.0
    Jinja2                         3.0.1
    lxml                           4.9.1
    matplotlib                     3.5.1
    mmcv                           1.2.7
    moxing-framework               2.1.0.5d9c87c8
    numpy                          1.19.5
    opencv-python                  4.1.2.30
    pandas                         1.1.5
    Pillow                         9.2.0
    pip                            22.1.2
    protobuf                       3.20.1
    psutil                         5.8.0
    PyYAML                         5.1
    requests                       2.27.1
    scikit-learn                   0.22.1
    scipy                          1.5.2
    sklearn                        0.0
    tensorboard                    2.1.1
    tensorboardX                   2.0
    torch                          1.8.0
    torchtext                      0.5.0
    torchvision                    0.9.0
    tornado                        6.2
    tqdm                           4.64.0
    traitlets                      5.3.0
    typing_extensions              4.3.0
    urllib3                        1.26.10
    watchdog                       2.0.0
    wcwidth                        0.2.5
    Werkzeug                       2.1.2
    wheel                          0.37.1
    yapf                           0.32.0
    zipp                           3.8.0
    ...
  • Certain apt installation packages:
    apt
    ca-certificates
    cmake
    cuda
    curl
    ethtool
    fdisk
    ffmpeg
    g++
    gcc
    git
    gpg
    graphviz
    libsm6
    libxext6
    libopencv-dev
    libxrender-dev
    libatlas3-base
    libnuma-dev
    libcap-dev
    libssl-dev
    liblz-dev
    libbz2-dev
    liblzma-dev
    libboost-graph-dev
    libsndfile1
    libcurl4-openssl-dev
    libopenblas-base
    liblapack3
    libopenblas-dev
    libprotobuf-dev
    libleveldb-dev
    libsnappy-dev
    libhdf5-serial-dev
    liblapacke-dev
    libgflags-dev
    libgoogle-glog-dev
    liblmdb-dev
    libatlas-base-dev
    librdmacm1
    libcap2-bin
    libpq-dev
    mysql-common
    net-tools
    nginx
    openslide-tools
    openssh-client
    openssh-server
    openssh-sftp-server
    openssl
    protobuf-compiler
    redis-server
    redis-tools
    rpm
    tar
    tofrodos
    unzip
    vim
    wget
    zip
    zlib1g-dev
    ...

Engine Version 2: pytorch_1.8.2-cuda_11.1-py_3.7-ubuntu_18.04-x86_64

  • Image path: swr.{region}.myhuaweicloud.com/aip/pytorch_1_8:pytorch_1.8.2-cuda_11.1-py_3.7-ubuntu_18.04-x86_64-20220524162601-50d6a18
  • Image creation time: 20220524162601 (yyyy-mm-dd-hh-mm-ss)
  • Image system version: Ubuntu 18.04.4 LTS
  • CUDA: 11.1.1
  • cuDNN: 8.0.5.39
  • Path and version of the Python interpreter: /home/ma-user/anaconda3/envs/PyTorch-1.8.2/bin/python, python 3.7.10
  • Third-party package installation path: /home/ma-user/anaconda3/envs/PyTorch-1.8.2/lib/python3.7/site-packages
  • Certain pip installation packages:
    Cython 0.27.3
    mmcv 1.2.7
    easydict 1.9
    tensorboardX 2.0
    torch 1.8.2+cu111
    Flask 2.0.1
    pandas 1.1.5
    gunicorn 20.1.0
    PyYAML 5.1
    torchaudio 0.8.2
    Pillow 9.0.1
    psutil 5.8.0
    lxml 4.8.0
    matplotlib 3.5.1
    torchvision 0.9.2+cu111
    pip 21.2.2
    protobuf 3.20.1
    numpy 1.17.0
    opencv-python 4.1.2.30
    scikit-learn 0.22.1
    ...
  • Certain apt installation packages:
    apt
    ca-certificates
    cmake
    cuda
    curl
    ethtool
    fdisk
    ffmpeg
    g++
    gcc
    git
    gpg
    graphviz
    libsm6
    libxext6
    libopencv-dev
    libxrender-dev
    libatlas3-base
    libnuma-dev
    libcap-dev
    libssl-dev
    liblz-dev
    libbz2-dev
    liblzma-dev
    libboost-graph-dev
    libsndfile1
    libcurl4-openssl-dev
    libopenblas-base
    liblapack3
    libopenblas-dev
    libprotobuf-dev
    libleveldb-dev
    libsnappy-dev
    libhdf5-serial-dev
    liblapacke-dev
    libgflags-dev
    libgoogle-glog-dev
    liblmdb-dev
    libatlas-base-dev
    librdmacm1
    libcap2-bin
    libpq-dev
    mysql-common
    net-tools
    nginx
    openslide-tools
    openssh-client
    openssh-server
    openssh-sftp-server
    openssl
    protobuf-compiler
    redis-server
    redis-tools
    rpm
    tar
    tofrodos
    unzip
    vim
    wget
    zip
    zlib1g-dev
    ...