kumatea/pytorch

pytorch-aarch64

镜像概述

pytorch-aarch64是专为AArch64（ARM64）架构构建的Docker镜像，集成了PyTorch及其相关库（torchvision、torchaudio、torchtext），支持Python 3.6至3.10版本。该镜像旨在为ARM架构设备提供便捷的深度学习开发与部署环境，适用于模型训练、推理及相关研究工作。

核心功能与特性

• 架构支持：专为AArch64架构优化，适配ARM64设备
• 版本支持：
  • PyTorch版本：1.7.1、1.8.0、1.8.1、1.9.0（最新）
  • Python版本：3.6、3.7、3.8、3.9、3.10
• 标签类型：
  • 标准标签：基础PyTorch环境
  • "full"标签：包含更多依赖的完整环境
• 持续集成：通过CircleCI构建验证，确保镜像质量

适用场景

• ARM架构设备（如树莓派、Jetson系列、ARM服务器）上的深度学习开发
• AArch64平台下的PyTorch模型训练与推理
• 跨平台深度学习应用的容器化部署

使用方法

拉取镜像

从Docker Hub拉取最新版本（默认Python 3.9）：

bash
docker pull kumatea/pytorch:latest

拉取特定Python版本（如Python 3.8）：

bash
docker pull kumatea/pytorch:py38

拉取完整版本（包含更多依赖）：

bash
docker pull kumatea/pytorch:py39-full

运行容器

启动交互式容器（以Python 3.9版本为例）：

bash
docker run -it --rm kumatea/pytorch:py39 python

挂载本地目录并运行脚本：

bash
docker run -it --rm -v $(pwd):/workspace kumatea/pytorch:latest python /workspace/your_script.py

标签说明

主要标签（1.9.0版本）

• 1.9.0, 1.9.0-py39, py39, latest
  • 1.9.0-full, 1.9.0-py39-full, py39-full（完整版本）
• 1.9.0-py38, py38
  • 1.9.0-py38-full, py38-full（完整版本）
• 1.9.0-py37, py37
  • 1.9.0-py37-full, py37-full（完整版本）
• 1.9.0-py36, py36
  • 1.9.0-py36-full, py36-full（完整版本）
• 1.9.0-py310, py310
  • 1.9.0-py310-full, py310-full（完整版本）

旧版本标签

1.8.1

• 1.8.1, 1.8.1-py39, py39, latest
  • 1.8.1-full, 1.8.1-py39-full, py39-full
• 1.8.1-py38, py38
  • 1.8.1-py38-full, py38-full
• 1.8.1-py37, py37
  • 1.8.1-py37-full, py37-full
• 1.8.1-py36, py36
  • 1.8.1-py36-full, py36-full

1.8.0

• 1.8.0, 1.8.0-py39
  • 1.8.0-full, 1.8.0-py39-full
• 1.8.0-py38
  • 1.8.0-py38-full
• 1.8.0-py37
  • 1.8.0-py37-full
• 1.8.0-py36
  • 1.8.0-py36-full

1.7.1

• 1.7.1, 1.7.1-py39
  • 1.7.1-full, 1.7.1-py39-full
• 1.7.1-py38
  • 1.7.1-py38-full
• 1.7.1-py37
  • 1.7.1-py37-full
• 1.7.1-py36
  • 1.7.1-py36-full

https://hub.docker.com/r/kumatea/pytorch/tags