pytorch镜像ReleaseNotes

PyTorch是一种强大的开源机器学习框架,支持基本的数学和张量运算,用户能够使用CPU和GPU实现各种深度学习。pytorch镜像为用户提供开箱即用的深度学习环境。

PyTorch 2.3.0.3

镜像列表

镜像类别

镜像地址

CPU

ac2-registry.cn-hangzhou.cr.aliyuncs.com/ac2/pytorch:2.3.0.3-alinux3.2304-enhanced

GPU

ac2-registry.cn-hangzhou.cr.aliyuncs.com/ac2/pytorch:2.3.0.3-cuda12.1.1-alinux3.2304-enhanced

GPU

ac2-registry.cn-hangzhou.cr.aliyuncs.com/ac2/pytorch:2.3.0.3-cuda11.8.0-alinux3.2304-enhanced

镜像内容

pytorch:2.3.0.3-alinux3.2304-enhanced

  • gcc: 12.2.1

  • Python: 3.10.13

  • bcrypt: 3.2.2

  • certifi: 2023.7.22

  • cffi: 1.15.1

  • charset-normalizer: 3.1.0

  • coverage: 7.2.1

  • cryptography: 42.0.5

  • exceptiongroup: 1.1.1

  • filelock: 3.9.0

  • fsspec: 2023.6.0

  • idna: 3.7

  • iniconfig: 1.1.1

  • Jinja2: 3.1.2

  • libcomps: 0.1.19

  • lxml: 4.9.2

  • MarkupSafe: 2.1.2

  • mpmath: 1.3.0

  • networkx: 2.8.8

  • numpy: 1.24.2

  • olefile: 0.46

  • packaging: 23.0

  • paramiko: 2.12.0

  • Pillow: 10.1.0

  • pip: 23.3.1

  • pluggy: 1.0.0

  • ply: 3.11

  • pyarrow: 14.0.2

  • pyasn1: 0.4.8

  • pycparser: 2.21

  • PyNaCl: 1.4.0

  • PySocks: 1.7.1

  • pytest: 7.3.1

  • pytest-cov: 4.0.0

  • requests: 2.31.0

  • setuptools: 65.5.1

  • six: 1.16.0

  • sympy: 1.11.1

  • tomli: 2.0.1

  • torch: 2.3.0.3.post1

  • torchaudio: 2.3.0.3.post1

  • torchvision: 0.18.0.3.post1

  • tqdm: 4.65.0

  • typing_extensions: 4.9.0

  • urllib3: 1.26.18

pytorch:2.3.0.3-cuda12.1.1-alinux3.2304-enhanced

  • Python: 3.10.13

  • cuda-cudart-12-1: 12.1.1

  • cuda-libraries-12-1: 12.1.1

  • cuda-nvrtc-12-1: 12.1.105

  • cuda-nvtx-12-1: 12.1.105

  • cuda-opencl-12-1: 12.1.105

  • libnpp-12-1: 12.1.0.40

  • libcublas-12-1: 12.1.3.1

  • libcufft-12-1: 11.0.2.54

  • libcufile-12-1: 1.6.1.9

  • libcurand-12-1: 10.3.2.106

  • libcusolver-12-1: 11.4.5.107

  • libcuparse-12-1: 12.1.0.106

  • libnvjitlink-12-1: 12.1.105

  • libnvjpeg-12-1: 12.2.0.2

  • libnccl: 2.20.5

  • libibverbs: 44.0

  • libnl3: 3.7.0

  • librdmacm: 44.0

  • libcudnn: 8.9.3.28

  • cuda-cupti-12-1: 12.1.105

  • cuda-nvtx-12-1: 12.1.105

  • cuda-compat-12-1: 530.30.02

  • bcrypt: 3.2.2

  • certifi: 2023.7.22

  • cffi: 1.15.1

  • charset-normalizer: 3.1.0

  • coverage: 7.2.1

  • cryptography: 42.0.5

  • exceptiongroup: 1.1.1

  • filelock: 3.9.0

  • fsspec: 2023.6.0

  • idna: 3.7

  • iniconfig: 1.1.1

  • Jinja2: 3.1.2

  • libcomps: 0.1.19

  • lxml: 4.9.2

  • MarkupSafe: 2.1.2

  • mpmath: 1.3.0

  • networkx: 2.8.8

  • numpy: 1.24.2

  • packaging: 23.0

  • paramiko: 2.12.0

  • pip: 23.3.1

  • pluggy: 1.0.0

  • ply: 3.11

  • pyarrow: 14.0.2

  • pyasn1: 0.4.8

  • pycparser: 2.21

  • PyNaCl: 1.4.0

  • PySocks: 1.7.1

  • pytest: 7.3.1

  • pytest-cov: 4.0.0

  • requests: 2.31.0

  • setuptools: 65.5.1

  • six: 1.16.0

  • sympy: 1.11.1

  • tomli: 2.0.1

  • torch: 2.3.0.3.post1+cu121

  • torchaudio: 2.3.0.3.post1+cu121

  • torchvision: 0.18.0.3.post1+cu121

  • tqdm: 4.65.0

  • triton: 2.3.0

  • typing_extensions: 4.9.0

  • urllib3: 1.26.18

pytorch:2.3.0.3-cuda11.8.0-alinux3.2304-enhanced

  • Python: 3.10.13

  • cuda-cudart-11-8: 11.8.89

  • cuda-libraries-11-8: 11.8.0

  • cuda-nvtx-11-8: 11.8.86

  • cuda-nvrtc-11-8: 11.8.89

  • libnpp-11-8: 11.8.0.86

  • libcublas-11-8: 11.11.3.6

  • libcufft-11-8: 10.9.0.58

  • libcufile-11-8: 1.4.0.31

  • libcurand-11-8: 10.3.0.86

  • libcusolver-11-8: 11.4.1.48

  • libcusparse-11-8: 11.7.5.86

  • libnccl-cuda-11: 2.20.5

  • libnvjpeg-11-8: 11.8.0.86

  • libibverbs: 46.0

  • libnl3: 3.7.0

  • librdmacm: 46.0

  • libcudnn-cuda-11: 8.9.3.28

  • cuda-cupti-11-8: 11.8.87

  • cuda-nvtx-11-8: 11.8.86

  • bcrypt: 3.2.2

  • certifi: 2023.7.22

  • cffi: 1.15.1

  • charset-normalizer: 3.1.0

  • coverage: 7.2.1

  • cryptography: 42.0.5

  • exceptiongroup: 1.1.1

  • filelock: 3.9.0

  • fsspec: 2023.6.0

  • idna: 3.7

  • iniconfig: 1.1.1

  • Jinja2: 3.1.2

  • libcomps: 0.1.19

  • lxml: 4.9.2

  • MarkupSafe: 2.1.2

  • mpmath: 1.3.0

  • networkx: 2.8.8

  • numpy: 1.24.2

  • packaging: 23.0

  • paramiko: 2.12.0

  • pip: 23.3.1

  • pluggy: 1.0.0

  • ply: 3.11

  • pyarrow: 14.0.2

  • pyasn1: 0.4.8

  • pycparser: 2.21

  • PyNaCl: 1.4.0

  • PySocks: 1.7.1

  • pytest: 7.3.1

  • pytest-cov: 4.0.0

  • requests: 2.31.0

  • setuptools: 65.5.1

  • six: 1.16.0

  • sympy: 1.11.1

  • tomli: 2.0.1

  • torch: 2.3.0.3.post1+cu118

  • torchaudio: 2.3.0.3.post1+cu118

  • torchvision: 0.18.0.3.post1+cu118

  • tqdm: 4.65.0

  • triton: 2.3.0

  • typing_extensions: 4.9.0

  • urllib3: 1.26.18

镜像运行要求

  • pytorch:2.3.0.3-alinux3.2304为CPU镜像,对驱动无要求。

  • pytorch:2.3.0.3-cuda12.1.1-alinux3.2304镜像,包含CUDA 12.1.1,需要nvidia-driver >= 525。兼容nvidia-driver R470。

  • pytorch:2.3.0.3-cuda11.8.0-alinux3.2304镜像,需要nvidia-driver >= 470。

重要特性

  • pytorch版本更新至2.3.0.3.post1,torchvision和torchaudio同步更新。

  • pytorch 2.3.0.3.post1在pytorch 2.3.0基础上,优化dynamo的模型和性能支持,能够在使用deepspeed/fsdp的同时获得dynamo的性能优化。

更新记录

2024.07:发布2.3.0.3镜像,包含CPU,CUDA11.8,CUDA12.1镜像。

PyTorch 2.3.0

镜像列表

镜像类别

镜像地址

CPU

ac2-registry.cn-hangzhou.cr.aliyuncs.com/ac2/pytorch:2.3.0-alinux3.2304

GPU

ac2-registry.cn-hangzhou.cr.aliyuncs.com/ac2/pytorch:2.3.0-cuda12.1.1-alinux3.2304

GPU

ac2-registry.cn-hangzhou.cr.aliyuncs.com/ac2/pytorch:2.3.0-cuda11.8.0-alinux3.2304

镜像内容

pytorch:2.3.0-alinux3.2304

  • gcc: 12.2.1

  • Python: 3.10.13

  • bcrypt: 3.2.2

  • certifi: 2023.7.22

  • cffi: 1.15.1

  • charset-normalizer: 3.1.0

  • coverage: 7.2.1

  • cryptography: 42.0.5

  • exceptiongroup: 1.1.1

  • filelock: 3.9.0

  • fsspec: 2023.6.0

  • idna: 3.7

  • iniconfig: 1.1.1

  • Jinja2: 3.1.2

  • libcomps: 0.1.19

  • lxml: 4.9.2

  • MarkupSafe: 2.1.2

  • mpmath: 1.3.0

  • networkx: 2.8.8

  • numpy: 1.24.2

  • olefile: 0.46

  • packaging: 23.0

  • paramiko: 2.12.0

  • Pillow: 10.1.0

  • pip: 23.3.1

  • pluggy: 1.0.0

  • ply: 3.11

  • pyarrow: 14.0.2

  • pyasn1: 0.4.8

  • pycparser: 2.21

  • PyNaCl: 1.4.0

  • PySocks: 1.7.1

  • pytest: 7.3.1

  • pytest-cov: 4.0.0

  • requests: 2.31.0

  • setuptools: 65.5.1

  • six: 1.16.0

  • sympy: 1.11.1

  • tomli: 2.0.1

  • torch: 2.3.0

  • torchaudio: 2.3.0

  • torchvision: 0.18.0

  • tqdm: 4.65.0

  • typing_extensions: 4.9.0

  • urllib3: 1.26.18

pytorch:2.3.0-cuda12.1.1-alinux3.2304

  • Python: 3.10.13

  • cuda-cudart-12-1: 12.1.1

  • cuda-libraries-12-1: 12.1.1

  • cuda-nvrtc-12-1: 12.1.105

  • cuda-nvtx-12-1: 12.1.105

  • cuda-opencl-12-1: 12.1.105

  • libnpp-12-1: 12.1.0.40

  • libcublas-12-1: 12.1.3.1

  • libcufft-12-1: 11.0.2.54

  • libcufile-12-1: 1.6.1.9

  • libcurand-12-1: 10.3.2.106

  • libcusolver-12-1: 11.4.5.107

  • libcuparse-12-1: 12.1.0.106

  • libnvjitlink-12-1: 12.1.105

  • libnvjpeg-12-1: 12.2.0.2

  • libnccl: 2.20.5

  • libibverbs: 44.0

  • libnl3: 3.7.0

  • librdmacm: 44.0

  • libcudnn: 8.9.3.28

  • cuda-cupti-12-1: 12.1.105

  • cuda-nvtx-12-1: 12.1.105

  • cuda-compat-12-1: 530.30.02

  • bcrypt: 3.2.2

  • certifi: 2023.7.22

  • cffi: 1.15.1

  • charset-normalizer: 3.1.0

  • coverage: 7.2.1

  • cryptography: 42.0.5

  • exceptiongroup: 1.1.1

  • filelock: 3.9.0

  • fsspec: 2023.6.0

  • idna: 3.7

  • iniconfig: 1.1.1

  • Jinja2: 3.1.2

  • libcomps: 0.1.19

  • lxml: 4.9.2

  • MarkupSafe: 2.1.2

  • mpmath: 1.3.0

  • networkx: 2.8.8

  • numpy: 1.24.2

  • packaging: 23.0

  • paramiko: 2.12.0

  • pip: 23.3.1

  • pluggy: 1.0.0

  • ply: 3.11

  • pyarrow: 14.0.2

  • pyasn1: 0.4.8

  • pycparser: 2.21

  • PyNaCl: 1.4.0

  • PySocks: 1.7.1

  • pytest: 7.3.1

  • pytest-cov: 4.0.0

  • requests: 2.31.0

  • setuptools: 65.5.1

  • six: 1.16.0

  • sympy: 1.11.1

  • tomli: 2.0.1

  • torch: 2.3.0+cu121

  • torchaudio: 2.3.0+cu121

  • torchvision: 0.18.0+cu121

  • tqdm: 4.65.0

  • triton: 2.3.0

  • typing_extensions: 4.9.0

  • urllib3: 1.26.18

pytorch:2.3.0-cuda11.8.0-alinux3.2304

  • Python: 3.10.13

  • cuda-cudart-11-8: 11.8.89

  • cuda-libraries-11-8: 11.8.0

  • cuda-nvtx-11-8: 11.8.86

  • cuda-nvrtc-11-8: 11.8.89

  • libnpp-11-8: 11.8.0.86

  • libcublas-11-8: 11.11.3.6

  • libcufft-11-8: 10.9.0.58

  • libcufile-11-8: 1.4.0.31

  • libcurand-11-8: 10.3.0.86

  • libcusolver-11-8: 11.4.1.48

  • libcusparse-11-8: 11.7.5.86

  • libnccl-cuda-11: 2.20.5

  • libnvjpeg-11-8: 11.8.0.86

  • libibverbs: 46.0

  • libnl3: 3.7.0

  • librdmacm: 46.0

  • libcudnn-cuda-11: 8.9.3.28

  • cuda-cupti-11-8: 11.8.87

  • cuda-nvtx-11-8: 11.8.86

  • bcrypt: 3.2.2

  • certifi: 2023.7.22

  • cffi: 1.15.1

  • charset-normalizer: 3.1.0

  • coverage: 7.2.1

  • cryptography: 42.0.5

  • exceptiongroup: 1.1.1

  • filelock: 3.9.0

  • fsspec: 2023.6.0

  • idna: 3.7

  • iniconfig: 1.1.1

  • Jinja2: 3.1.2

  • libcomps: 0.1.19

  • lxml: 4.9.2

  • MarkupSafe: 2.1.2

  • mpmath: 1.3.0

  • networkx: 2.8.8

  • numpy: 1.24.2

  • packaging: 23.0

  • paramiko: 2.12.0

  • pip: 23.3.1

  • pluggy: 1.0.0

  • ply: 3.11

  • pyarrow: 14.0.2

  • pyasn1: 0.4.8

  • pycparser: 2.21

  • PyNaCl: 1.4.0

  • PySocks: 1.7.1

  • pytest: 7.3.1

  • pytest-cov: 4.0.0

  • requests: 2.31.0

  • setuptools: 65.5.1

  • six: 1.16.0

  • sympy: 1.11.1

  • tomli: 2.0.1

  • torch: 2.3.0+cu118

  • torchaudio: 2.3.0+cu118

  • torchvision: 0.18.0+cu118

  • tqdm: 4.65.0

  • triton: 2.3.0

  • typing_extensions: 4.9.0

  • urllib3: 1.26.18

镜像运行要求

  • pytorch:2.3.0-alinux3.2304为CPU镜像,对驱动无要求。

  • pytorch:2.3.0-cuda12.1.1-alinux3.2304镜像,包含CUDA 12.1.1,需要nvidia-driver >= 525。兼容nvidia-driver R470。

  • pytorch:2.3.0-cuda11.8.0-alinux3.2304镜像,需要nvidia-driver >= 470。

重要特性

pytorch版本更新至2.3.0,torchvision和torchaudio同步更新。

更新记录

2024.06:发布2.3.0镜像,包含CPU,CUDA11.8,CUDA12.1镜像。

PyTorch 2.2.0.1

镜像列表

镜像类别

镜像地址

GPU

ac2-registry.cn-hangzhou.cr.aliyuncs.com/ac2/pytorch:2.2.0.1-3.2304-cu121

GPU

ac2-registry.cn-hangzhou.cr.aliyuncs.com/ac2/pytorch:2.2.0.1-3.2304-cu118

CPU

ac2-registry.cn-hangzhou.cr.aliyuncs.com/ac2/pytorch:2.2.0.1-3.2304

镜像内容

pytorch:2.2.0.1-3.2304

  • gcc: 12.2.1

  • Python: 3.10.13

  • aiodns: 3.0.0

  • aiohttp: 3.8.4

  • aiosignal: 1.3.1

  • appdirs: 1.4.4

  • async-timeout: 4.0.2

  • attrs: 22.2.0

  • bcrypt: 3.2.2

  • Brotli: 1.0.9

  • certifi: 2023.7.22

  • cffi: 1.15.1

  • charset-normalizer: 3.1.0

  • contourpy: 1.0.7

  • coverage: 7.2.1

  • cryptography: 39.0.0

  • cycler: 0.11.0

  • exceptiongroup: 1.1.1

  • filelock: 3.9.0

  • fonttools: 4.39.2

  • frozenlist: 1.3.3

  • fs: 2.4.16

  • fsspec: 2023.6.0

  • idna: 3.4

  • iniconfig: 1.1.1

  • Jinja2: 3.1.2

  • kiwisolver: 1.4.4

  • libcomps: 0.1.19

  • lxml: 4.9.2

  • MarkupSafe: 2.1.2

  • matplotlib: 3.7.1

  • mpmath: 1.3.0

  • multidict: 6.0.4

  • munkres: 1.1.2

  • networkx: 2.8.8

  • numpy: 1.24.2

  • olefile: 0.46

  • packaging: 23.0

  • paramiko: 2.12.0

  • Pillow: 9.4.0

  • pip: 23.3.1

  • pluggy: 1.0.0

  • ply: 3.11

  • pyarrow: 14.0.2

  • pyasn1: 0.4.8

  • pycairo: 1.23.0

  • pycares: 4.3.0

  • pycparser: 2.21

  • PyNaCl: 1.4.0

  • pyparsing: 3.0.7

  • PySocks: 1.7.1

  • pytest: 7.3.1

  • pytest-cov: 4.0.0

  • python-dateutil: 2.8.2

  • requests: 2.31.0

  • SciPy: 1.10.1

  • setuptools: 65.5.1

  • six: 1.16.0

  • sympy: 1.11.1

  • tomli: 2.0.1

  • torch: 2.2.0.1

  • torchaudio: 2.2.0.1

  • torchvision: 0.17.0.1

  • tqdm: 4.65.0

  • typing_extensions: 4.9.0

  • urllib3: 1.26.18

  • yarl: 1.8.2

pytorch:2.2.0.1-3.2304-cu121

  • Python: 3.10.13

  • cuda-cudart-12-1: 12.1.1

  • cuda-libraries-12-1: 12.1.1

  • cuda-nvrtc-12-1: 12.1.105

  • cuda-nvtx-12-1: 12.1.105

  • cuda-opencl-12-1: 12.1.105

  • libnpp-12-1: 12.1.0.40

  • libcublas-12-1: 12.1.3.1

  • libcufft-12-1: 11.0.2.54

  • libcufile-12-1: 1.6.1.9

  • libcurand-12-1: 10.3.2.106

  • libcusolver-12-1: 11.4.5.107

  • libcuparse-12-1: 12.1.0.106

  • libnvjitlink-12-1: 12.1.105

  • libnvjpeg-12-1: 12.2.0.2

  • libnccl: 2.19.3

  • libibverbs: 44.0

  • libnl3: 3.7.0

  • librdmacm: 44.0

  • libcudnn: 8.9.3.28

  • cuda-cupti-12-1: 12.1.105

  • cuda-nvtx-12-1: 12.1.105

  • cuda-compat-12-1: 530.30.02

  • aiodns: 3.0.0

  • aiohttp: 3.8.4

  • aiosignal: 1.3.1

  • appdirs: 1.4.4

  • async-timeout: 4.0.2

  • attrs: 22.2.0

  • bcrypt: 3.2.2

  • Brotli: 1.0.9

  • certifi: 2023.7.22

  • cffi: 1.15.1

  • charset-normalizer: 3.1.0

  • contourpy: 1.0.7

  • coverage: 7.2.1

  • cryptography: 39.0.0

  • cycler: 0.11.0

  • exceptiongroup: 1.1.1

  • filelock: 3.9.0

  • fonttools: 4.39.2

  • frozenlist: 1.3.3

  • fs: 2.4.16

  • fsspec: 2023.6.0

  • idna: 3.4

  • iniconfig: 1.1.1

  • Jinja2: 3.1.2

  • kiwisolver: 1.4.4

  • libcomps: 0.1.19

  • lxml: 4.9.2

  • MarkupSafe: 2.1.2

  • matplotlib: 3.7.1

  • mpmath: 1.3.0

  • multidict: 6.0.4

  • munkres: 1.1.2

  • networkx: 2.8.8

  • numpy: 1.24.2

  • olefile: 0.46

  • packaging: 23.0

  • paramiko: 2.12.0

  • Pillow: 9.4.0

  • pip: 23.3.1

  • pluggy: 1.0.0

  • ply: 3.11

  • pyarrow: 14.0.2

  • pyasn1: 0.4.8

  • pycairo: 1.23.0

  • pycares: 4.3.0

  • pycparser: 2.21

  • PyNaCl: 1.4.0

  • pyparsing: 3.0.7

  • PySocks: 1.7.1

  • pytest: 7.3.1

  • pytest-cov: 4.0.0

  • python-dateutil: 2.8.2

  • requests: 2.31.0

  • SciPy: 1.10.1

  • setuptools: 65.5.1

  • six: 1.16.0

  • sympy: 1.11.1

  • tomli: 2.0.1

  • torch: 2.2.0.1+cu121

  • torchaudio: 2.2.0.1+cu121

  • torchvision: 0.17.0.1+cu121

  • tqdm: 4.65.0

  • triton: 2.2.0

  • typing_extensions: 4.9.0

  • urllib3: 1.26.18

  • yarl: 1.8.2

pytorch:2.2.0.1-3.2304-cu118

  • Python: 3.10.13

  • cuda-cudart-11-8: 11.8.89

  • cuda-libraries-11-8: 11.8.0

  • cuda-nvtx-11-8: 11.8.86

  • cuda-nvrtc-11-8: 11.8.89

  • libnpp-11-8: 11.8.0.86

  • libcublas-11-8: 11.11.3.6

  • libcufft-11-8: 10.9.0.58

  • libcufile-11-8: 1.4.0.31

  • libcurand-11-8: 10.3.0.86

  • libcusolver-11-8: 11.4.1.48

  • libcusparse-11-8: 11.7.5.86

  • libnccl-cuda-11: 2.19.3

  • libnvjpeg-11-8: 11.8.0.86

  • libibverbs: 46.0

  • libnl3: 3.7.0

  • librdmacm: 46.0

  • libcudnn-cuda-11: 8.9.3.28

  • cuda-cupti-11-8: 11.8.87

  • cuda-nvtx-11-8: 11.8.86

  • aiodns: 3.0.0

  • aiohttp: 3.8.4

  • aiosignal: 1.3.1

  • appdirs: 1.4.4

  • async-timeout: 4.0.2

  • attrs: 22.2.0

  • bcrypt: 3.2.2

  • Brotli: 1.0.9

  • certifi: 2023.7.22

  • cffi: 1.15.1

  • charset-normalizer: 3.1.0

  • contourpy: 1.0.7

  • coverage: 7.2.1

  • cryptography: 39.0.0

  • cycler: 0.11.0

  • exceptiongroup: 1.1.1

  • filelock: 3.9.0

  • fonttools: 4.47.0

  • frozenlist: 1.3.3

  • fs: 2.4.16

  • fsspec: 2023.6.0

  • idna: 3.4

  • iniconfig: 1.1.1

  • Jinja2: 3.1.2

  • kiwisolver: 1.4.4

  • libcomps: 0.1.19

  • lxml: 4.9.2

  • MarkupSafe: 2.1.2

  • matplotlib: 3.7.1

  • mpmath: 1.3.0

  • multidict: 6.0.4

  • networkx: 2.8.8

  • numpy: 1.24.2

  • olefile: 0.46

  • packaging: 23.0

  • paramiko: 2.12.0

  • Pillow: 9.4.0

  • pip: 23.3.1

  • pluggy: 1.0.0

  • ply: 3.11

  • pyarrow: 14.0.2

  • pyasn1: 0.4.8

  • pycairo: 1.23.0

  • pycares: 4.3.0

  • pycparser: 2.21

  • PyNaCl: 1.4.0

  • pyparsing: 3.0.7

  • PySocks: 1.7.1

  • pytest: 7.3.1

  • pytest-cov: 4.0.0

  • python-dateutil: 2.8.2

  • requests: 2.31.0

  • SciPy: 1.10.1

  • setuptools: 65.5.1

  • six: 1.16.0

  • sympy: 1.11.1

  • tomli: 2.0.1

  • torch: 2.2.0.1+cu118

  • torchaudio: 2.2.0.1+cu118

  • torchvision: 0.17.0.1+cu118

  • tqdm: 4.65.0

  • triton: 2.2.0+cu118

  • typing_extensions: 4.9.0

  • urllib3: 1.26.18

  • yarl: 1.8.2

镜像运行要求

  • pytorch:2.2.0.1-alinux3.2304为CPU镜像,对驱动无要求。

  • pytorch:2.2.0.1-cuda12.1.1-alinux3.2304镜像,包含CUDA 12.1.1,需要nvidia-driver >= 525。兼容nvidia-driver R470。

  • pytorch:2.2.0.1-cuda11.8.0-alinux3.2304镜像,需要nvidia-driver >= 470。

重要特性

  • pytorch版本更新至2.2.0.1,torchvision和torchaudio同步更新。

  • pytorch 2.2.0.1在pytorch 2.2.0基础上,优化dynamo的模型和性能支持,能够在使用deepspeed/fsdp的同时获得dynamo的性能优化。

更新记录

2024.03:发布pytorch 2.2.0.1 CPU,CUDA11.8,CUDA12.1镜像。

PyTorch 2.1.0

镜像列表

镜像类别

镜像地址

DCU

ac2-registry.cn-hangzhou.cr.aliyuncs.com/ac2/pytorch:2.1.0-dtk24.04.1-py310-alinux3.2304

镜像内容

pytorch:2.1.0-dtk24.04.1-py310-alinux3.2304

  • Python: 3.10.13

  • DTK: 24.04.1

  • apex: 1.1.0+das1.1.gitf477a3a.abi1.dtk2404.torch2.1.0

  • certifi: 2024.6.2

  • charset-normalizer: 3.3.2

  • dropout-layer-norm: 0.1+das1.1gitc7a8c18.abi1.dtk2404.torch2.1

  • einops: 0.8.0

  • filelock: 3.15.4

  • flash-attn: 2.0.4+das1.1gitc7a8c18.abi1.dtk2404.torch2.1

  • fsspec: 2024.6.0

  • fused-dense-lib: 0.1+das1.1gitc7a8c18.abi1.dtk2404.torch2.1

  • idna: 3.7

  • Jinja2: 3.1.4

  • MarkupSafe: 2.1.5

  • mpmath: 1.3.0

  • networkx: 3.3

  • ninja: 1.11.1.1

  • numpy: 1.24.3

  • packaging: 24.0

  • pillow: 10.3.0

  • pip: 24.0

  • requests: 2.31.0

  • rotary-emb: 0.1+das1.1gitc7a8c18.abi1.dtk2404.torch2.1

  • setuptools: 68.2.2

  • sympy: 1.12.1

  • torch: 2.1.0+das1.1.git3ac1bdd.abi1.dtk2404

  • torchaudio: 2.1.2+das1.1.git63d9a68.abi1.dtk2404.torch2.1.0

  • torchvision: 0.16.0+das1.1.git7d45932.abi1.dtk2404.torch2.1

  • triton: 2.1.0+das1.1.git4bf1007a.abi1.dtk2404.torch2.1.0

  • typing_extensions: 4.12.2

  • urllib3: 2.2.2

  • xentropy-cuda-lib: 0.1+das1.1gitc7a8c18.abi1.dtk2404.torch2.1

  • xformers: 0.0.25+das1.1.git8ef8bc1.abi1.dtk2404.torch2.1.0

镜像运行要求

pytorch:2.1.0-dtk24.04.1-py310-alinux3.2304 需要驱动DTK 24.04版本,驱动安装方式参考DCU快速上手

重要特性

pytorch:2.1.0-dtk24.04.1-py310-alinux3.2304提供pytorch 2.1.0,torchaudio 2.1.2,torchvision 0.16.0,torch系列组件均从光合开发社区下载。

更新记录

2024.08:发布pytorch DTK镜像,DTK版本为24.04.1。

PyTorch 2.0.1

镜像列表

镜像类别

镜像地址

GPU

ac2-registry.cn-hangzhou.cr.aliyuncs.com/ac2/pytorch:2.0.1-3.2304

镜像内容

pytorch:2.0.1-3.2304

  • Python: 3.10.13

  • cuda-cudart-12-1: 12.1.105

  • cuda-compat-12-1: 530.30.2

  • libnvjitlink-12-1: 12.1.105

  • libcusparse-12-1: 12.1.0.106

  • libcublas-12-1: 12.1.3.1

  • libcusolver-12-1: 11.4.5.107

  • libcufile-12-1: 1.6.1.9

  • libnvjpeg-12-1: 12.2.0.2

  • libnpp-12-1: 12.1.0.40

  • libcurand-12-1: 10.3.2.106

  • libcufft-12-1: 11.0.2.54

  • cuda-opencl-12-1: 12.1.105

  • cuda-nvrtc-12-1: 12.1.105

  • cuda-libraries-12-1: 12.1.1

  • cuda-nvtx-12-1: 12.1.105

  • cuda-cupti-12-1: 12.1.105

  • libnccl: 2.18.3

  • libcudnn: 8.9.3.28

  • appdirs: 1.4.4

  • Brotli: 1.0.9

  • certifi: 2023.7.22

  • charset-normalizer: 3.1.0

  • contourpy: 1.0.7

  • cycler: 0.11.0

  • filelock: 3.9.0

  • fonttools: 4.39.2

  • fs: 2.4.16

  • idna: 3.4

  • Jinja2: 3.1.2

  • kiwisolver: 1.4.4

  • libcomps: 0.1.19

  • lxml: 4.9.2

  • MarkupSafe: 2.1.2

  • matplotlib: 3.7.1

  • mpmath: 1.3.0

  • munkres: 1.1.2

  • networkx: 2.8.8

  • numpy: 1.24.2

  • olefile: 0.46

  • packaging: 23.0

  • Pillow: 9.4.0

  • pip: 23.3.1

  • pycairo: 1.23.0

  • pyparsing: 3.0.7

  • PySocks: 1.7.1

  • python-dateutil: 2.8.2

  • requests: 2.31.0

  • SciPy: 1.10.1

  • setuptools: 65.5.1

  • six: 1.16.0

  • sympy: 1.11.1

  • torch: 2.0.1

  • torchvision: 0.15.2

  • typing_extensions: 4.9.0

  • urllib3: 1.26.18

镜像运行要求

pytorch:2.0.1 包含cuda 12.1.1,需要nvidia-driver >= 525。兼容nvidia-driver R470。

重要特性

该镜像包含pytorch 2.1.0和torchvision 0.15.2。

更新记录

2024.02:发布pytorch 2.0.1 CUDA12.1镜像。