PyTorch是一种强大的开源机器学习框架，支持基本的数学和张量运算，用户能够使用CPU和GPU实现各种深度学习。PyTorch镜像为用户提供开箱即用的深度学习环境。
PyTorch 2.5.1.6
镜像列表
镜像类别 | 镜像地址 |
AMD GPU | ac2-registry.cn-hangzhou.cr.aliyuncs.com/ac2/pytorch:2.5.1.6-rocm6.2.4-py310-alinux3.2104-enhanced |
GPU | ac2-registry.cn-hangzhou.cr.aliyuncs.com/ac2/pytorch:2.5.1.6-cuda12.1.1-py310-alinux3.2104 |
镜像内容
ac2-registry.cn-hangzhou.cr.aliyuncs.com/ac2/pytorch:2.5.1.6-rocm6.2.4-py310-alinux3.2104-enhanced
amd-smi-lib: 24.6.3.60204
amdgpu-core: 6.2.60204
half: 1.12.0.60204
hip-runtime-amd: 6.2.41134.60204
hipblas: 2.2.0.60204
hipblaslt: 0.8.0.60204
hipcc: 1.1.1.60204
hipfft: 1.0.16.60204
hipify-clang: 18.0.0.60204
hiprand: 2.11.1.60204
hipsolver: 2.2.0.60204
hipsparse: 3.1.1.60204
hipsparselt: 0.2.1.60204
hiptensor: 1.3.0.60204
hsa-rocr: 1.14.0.60204
miopen-hip: 3.2.0.60204
omnitrace: 1.11.2.60204
openmp-extras-runtime: 18.62.0.60204
rccl: 2.20.5.60204
rocalution: 3.2.1.60204
rocblas: 4.2.4.60204
rocfft: 1.0.30.60204
rocm-cmake: 0.13.0.60204
rocm-dbgapi: 0.76.0.60204
rocm-debug-agent: 2.0.3.60204
rocm-device-libs: 1.0.0.60204
rocm-gdb: 14.2.60204
rocm-llvm: 18.0.0.24392.60204
rocm-opencl-icd-loader: 1.2.60204
rocm-opencl: 2.0.0.60204
rocm-smi-lib: 7.3.0.60204
rocminfo: 1.0.0.60204
rocprim-devel: 3.2.2.60204
rocprofiler-plugins: 2.0.60204.60204
rocprofiler-register: 0.4.0.60204
rocprofiler: 2.0.60204.60204
rocrand: 3.1.1.60204
rocsolver: 3.26.2.60204
rocsparse: 3.2.1.60204
roctracer: 4.1.60204.60204
rocwmma-devel: 1.5.0.60204
Python: 3.10.13
jinja2: 3.1.4
networkx: 3.3
sympy: 1.13.1
fsspec: 2024.6.0
filelock: 3.15.4
typing-extensions: 4.12.2
certifi: 2024.12.14
charset-normalizer: 3.3.2
idna: 3.7
pysocks: 1.7.1
requests: 2.32.3
urllib3: 2.2.2
numpy: 1.26.4
pillow: 10.3.0
torch: 2.5.1.6
torchaudio: 2.5.1.6
torchvision: 0.20.1.6
pytorch_triton_rocm: 3.1.0
hjson: 3.0.2
ninja: 1.11.1.1
packaging: 24.0
psutil: 5.9.2
py_cpuinfo: 9.0.0
pydantic: 2.8.2
tqdm: 4.66.4
annotated_types: 0.7.0
pydantic_core: 2.20.1
accelerate: 1.2.1
transformers: 4.47.1
PyYAML: 6.0.1
huggingface_hub: 0.27.0
safetensors: 0.4.3
regex: 2024.5.15
tokenizers: 0.21.0
msgpack: 1.1.0
ac2-registry.cn-hangzhou.cr.aliyuncs.com/ac2/pytorch:2.5.1.6-cuda12.1.1-py310-alinux3.2104
cuda-cudart-12-1: 12.1.1
cuda-cupti-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
libcudnn9: 9.2.1.18
Python: 3.10.13
accelerate: 1.2.1
annotated-types: 0.7.0
certifi: 2024.12.14
charset-normalizer: 3.3.2
deepspeed: 0.15.4+ali
filelock: 3.15.4
fsspec: 2024.6.0
hjson: 3.0.2
huggingface_hub: 0.27.0
idna: 3.7
Jinja2: 3.1.4
MarkupSafe: 2.1.5
mpmath: 1.3.0
msgpack: 1.1.0
networkx: 3.3
ninja: 1.11.1.1
numpy: 1.26.4
packaging: 24.0
pillow: 10.3.0
pip: 24.0
psutil: 5.9.2
py-cpuinfo: 9.0.0
pydantic: 2.8.2
pydantic_core: 2.20.1
PySocks: 1.7.1
PyYAML: 6.0.1
regex: 2024.5.15
requests: 2.32.3
safetensors: 0.4.3
setuptools: 73.0.1
sympy: 1.13.1
tokenizers: 0.21.0
torch: 2.5.1.6+cu121
torchaudio: 2.5.1.6+cu121
torchvision: 0.20.1.6+cu121
tqdm: 4.66.4
transformers: 4.47.1
triton: 3.1.0
typing_extensions: 4.12.2
urllib3: 2.2.2
wheel: 0.43.0
镜像运行要求
pytorch:2.5.1.6-rocm6.2.4-py310-alinux3.2104-enhanced镜像包含rocm 6.2.4，参考User and kernel-space support matrix说明，需要host驱动版本不低于6.0.x。
pytorch:2.5.1.6-cuda12.1.1-py310-alinux3.2104镜像包含CUDA 12.1.1 RunTime，需要nvidia-drive版本不低于525，并且兼容nvidia-driver R470。
重要特性
镜像提供torch 2.5.1.6，以及配套版本的Torchvision和Torchaudio。
镜像中内置的Python 3版本为
3.10.13+gc，位于/opt/ac2/bin目录下，集成了Python3的GC优化功能，能够有效缓解在AI场景下出现的GC暂停问题。
更新记录
2025.01：发布pytorch 2.5.1.6 for rocm6.2.4镜像。
2025.02：发布pytorch 2.5.1.6 for cuda12.1.1镜像。
PyTorch 2.4.0
镜像列表
镜像类别 | 镜像地址 |
GPU | ac2-registry.cn-hangzhou.cr.aliyuncs.com/ac2/pytorch:2.4.0-cuda12.1.1-py310-alinux3.2104 |
镜像内容
ac2-registry.cn-hangzhou.cr.aliyuncs.com/ac2/pytorch:2.4.0-cuda12.1.1-py310-alinux3.2104
Python: 3.10.13
cuda-cudart-12-1: 12.1.1
cuda-cupti-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
libcudnn9: 9.2.1.18
jinja2: 3.1.4
networkx: 3.3
sympy: 1.12.1
fsspec: 2024.6.0
filelock: 3.15.4
typing-extensions: 4.12.2
certifi: 2024.6.2
charset-normalizer: 3.3.2
idna: 3.7
pysocks: 1.7.1
requests: 2.32.3
urllib3: 2.2.2
numpy: 1.26.4
pillow: 10.3.0
torch: 2.4.0+cu121
torchaudio: 2.4.0+cu121
torchvision: 0.19.0+cu121
镜像运行要求
pytorch:2.4.0-cuda12.1.1-py310-alinux3.2104镜像包含CUDA 12.1.1，需要nvidia-driver版本不低于525，并且兼容nvidia-driver R470。
重要特性
PyTorch版本更新至2.4.0，Torchvision和Torchaudio同步更新。
更新记录
2024.11：发布2.4.0 CUDA12.1.1镜像。
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、CUDA 11.8及CUDA 12.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、CUDA 11.8和CUDA 12.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镜像。