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
View PDF

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
...
Parent topic: Inference Base Images