Pytorch Docker 镜像

简体中文 | English

Note

这是一个从 cnstark/pytorch-docker 派生的项目，支持几乎各种排列组合 pytorch 1.4-2.5、python 3.8-3.12、ubuntu 18.04-22.04、centos 8 和 cuda 10.0-12.4。

提供不同操作系统、CUDA 和 Python 版本的纯 Pytorch Docker 镜像。

Github: https://github.com/xiongsp/pytorch-docker

Docker Hub: https://hub.docker.com/r/spxiong/pytorch

更新日志

2024/12/01: 支持 pytorch 1.4-2.0。
2024/11/05: 支持 Pytorch 2.5.x。对于 2.1 及以下版本，请参考此 DockerHub 和 Github 仓库。
2024/10/10: 支持 Pytorch 2.1.x 2.2.x 2.3.x 2.4.x。

使用方法

准备工作

Docker
Nvidia-docker（如果需要 CUDA）

使用 Pytorch Docker

与 Pytorch 官方相同。

docker run -it --rm \
    --gpus all \
    --net host
    -v /path/to/project:/path/to/project \
    -v /path/to/dataset:/path/to/dataset \
    spxiong/pytorch:[TAG]

镜像列表

2.5.1

Image	Pull Command
	`docker pull spxiong/pytorch:2.5.1-py3.10.15-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.1-py3.10.15-cuda12.1.0-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.1-py3.10.15-cuda12.1.0-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.1-py3.10.15-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.1-py3.10.15-cuda12.4.1-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.1-py3.10.15-cuda12.4.1-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.1-py3.11.10-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.1-py3.11.10-cuda12.1.0-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.1-py3.11.10-cuda12.1.0-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.1-py3.11.10-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.1-py3.11.10-cuda12.4.1-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.1-py3.11.10-cuda12.4.1-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.1-py3.12.7-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.1-py3.12.7-cuda12.1.0-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.1-py3.12.7-cuda12.1.0-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.1-py3.12.7-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.1-py3.12.7-cuda12.4.1-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.1-py3.12.7-cuda12.4.1-devel-ubuntu22.04`

2.5.0

Image	Pull Command
	`docker pull spxiong/pytorch:2.5.0-py3.10.15-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.0-py3.10.15-cuda12.1.0-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.0-py3.10.15-cuda12.1.0-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.0-py3.10.15-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.0-py3.10.15-cuda12.4.1-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.0-py3.10.15-cuda12.4.1-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.0-py3.11.10-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.0-py3.11.10-cuda12.1.0-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.0-py3.11.10-cuda12.1.0-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.0-py3.11.10-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.0-py3.11.10-cuda12.4.1-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.0-py3.11.10-cuda12.4.1-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.0-py3.12.7-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.0-py3.12.7-cuda12.1.0-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.0-py3.12.7-cuda12.1.0-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.0-py3.12.7-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.0-py3.12.7-cuda12.4.1-ubuntu22.04`
	`docker pull spxiong/pytorch:2.5.0-py3.12.7-cuda12.4.1-devel-ubuntu22.04`

2.1-2.4

2.4.1

Image	Pull Command
	`docker pull spxiong/pytorch:2.4.1-py3.10.15-cuda12.1.0-ubuntu22.04`
	`docker pull spxiong/pytorch:2.4.1-py3.10.15-cuda12.1.0-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.4.1-py3.10.15-ubuntu22.04`

2.4.0

Image	Pull Command
	`docker pull spxiong/pytorch:2.4.0-py3.10.15-cuda12.1.0-ubuntu22.04`
	`docker pull spxiong/pytorch:2.4.0-py3.10.15-cuda12.1.0-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.4.0-py3.10.15-ubuntu22.04`

2.3.1

Image	Pull Command
	`docker pull spxiong/pytorch:2.3.1-py3.10.15-cuda12.1.0-ubuntu22.04`
	`docker pull spxiong/pytorch:2.3.1-py3.10.15-cuda12.1.0-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.3.1-py3.10.15-ubuntu22.04`

2.3.0

Image	Pull Command
	`docker pull spxiong/pytorch:2.3.0-py3.10.15-cuda12.1.0-ubuntu22.04`
	`docker pull spxiong/pytorch:2.3.0-py3.10.15-cuda12.1.0-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.3.0-py3.10.15-ubuntu22.04`

2.2.2

Image	Pull Command
	`docker pull spxiong/pytorch:2.2.2-py3.10.15-cuda12.1.0-ubuntu22.04`
	`docker pull spxiong/pytorch:2.2.2-py3.10.15-cuda12.1.0-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.2.2-py3.10.15-ubuntu22.04`

2.2.1

Image	Pull Command
	`docker pull spxiong/pytorch:2.2.1-py3.10.15-cuda12.1.0-ubuntu22.04`
	`docker pull spxiong/pytorch:2.2.1-py3.10.15-cuda12.1.0-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.2.1-py3.10.15-ubuntu22.04`

2.2.0

Image	Pull Command
	`docker pull spxiong/pytorch:2.2.0-py3.10.15-cuda12.1.0-ubuntu22.04`
	`docker pull spxiong/pytorch:2.2.0-py3.10.15-cuda12.1.0-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.2.0-py3.10.15-ubuntu22.04`

2.1.2

Image	Pull Command
	`docker pull spxiong/pytorch:2.1.2-py3.10.15-cuda12.1.0-ubuntu22.04`
	`docker pull spxiong/pytorch:2.1.2-py3.10.15-cuda12.1.0-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.1.2-py3.10.15-ubuntu22.04`

2.1.1

Image	Pull Command
	`docker pull spxiong/pytorch:2.1.1-py3.10.15-cuda12.1.0-ubuntu22.04`
	`docker pull spxiong/pytorch:2.1.1-py3.10.15-cuda12.1.0-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.1.1-py3.10.15-ubuntu22.04`

2.1.0

Image	Pull Command
	`docker pull spxiong/pytorch:2.1.0-py3.10.15-cuda12.1.0-ubuntu22.04`
	`docker pull spxiong/pytorch:2.1.0-py3.10.15-cuda12.1.0-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.1.0-py3.10.15-ubuntu22.04`

<= 2.0

2.0.1

Image	Pull Command
	`docker pull spxiong/pytorch:2.0.1-py3.9.12-ubuntu22.04`
	`docker pull spxiong/pytorch:2.0.1-py3.9.12-cuda11.7.1-ubuntu22.04`
	`docker pull spxiong/pytorch:2.0.1-py3.9.12-cuda11.7.1-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.0.1-py3.9.12-cuda11.8.0-ubuntu22.04`
	`docker pull spxiong/pytorch:2.0.1-py3.9.12-cuda11.8.0-devel-ubuntu22.04`

2.0.0

Image	Pull Command
	`docker pull spxiong/pytorch:2.0.0-py3.9.12-ubuntu22.04`
	`docker pull spxiong/pytorch:2.0.0-py3.9.12-cuda11.7.1-ubuntu22.04`
	`docker pull spxiong/pytorch:2.0.0-py3.9.12-cuda11.7.1-devel-ubuntu22.04`
	`docker pull spxiong/pytorch:2.0.0-py3.9.12-cuda11.8.0-ubuntu22.04`
	`docker pull spxiong/pytorch:2.0.0-py3.9.12-cuda11.8.0-devel-ubuntu22.04`

1.13.1

Image	Pull Command
	`docker pull spxiong/pytorch:1.13.1-py3.8.13-ubuntu20.04`
	`docker pull spxiong/pytorch:1.13.1-py3.8.13-cuda11.6.2-ubuntu20.04`
	`docker pull spxiong/pytorch:1.13.1-py3.8.13-cuda11.6.2-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.13.1-py3.8.13-cuda11.7.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.13.1-py3.8.13-cuda11.7.1-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.13.1-py3.9.12-ubuntu20.04`
	`docker pull spxiong/pytorch:1.13.1-py3.9.12-cuda11.6.2-ubuntu20.04`
	`docker pull spxiong/pytorch:1.13.1-py3.9.12-cuda11.6.2-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.13.1-py3.9.12-cuda11.7.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.13.1-py3.9.12-cuda11.7.1-devel-ubuntu20.04`

1.13.0

Image	Pull Command
	`docker pull spxiong/pytorch:1.13.0-py3.8.13-ubuntu20.04`
	`docker pull spxiong/pytorch:1.13.0-py3.8.13-cuda11.6.2-ubuntu20.04`
	`docker pull spxiong/pytorch:1.13.0-py3.8.13-cuda11.6.2-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.13.0-py3.8.13-cuda11.7.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.13.0-py3.8.13-cuda11.7.1-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.13.0-py3.9.12-ubuntu20.04`
	`docker pull spxiong/pytorch:1.13.0-py3.9.12-cuda11.6.2-ubuntu20.04`
	`docker pull spxiong/pytorch:1.13.0-py3.9.12-cuda11.6.2-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.13.0-py3.9.12-cuda11.7.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.13.0-py3.9.12-cuda11.7.1-devel-ubuntu20.04`

1.12.1

Image	Pull Command
	`docker pull spxiong/pytorch:1.12.1-py3.8.13-ubuntu20.04`
	`docker pull spxiong/pytorch:1.12.1-py3.8.13-cuda11.3.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.12.1-py3.8.13-cuda11.3.1-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.12.1-py3.8.13-cuda11.6.2-ubuntu20.04`
	`docker pull spxiong/pytorch:1.12.1-py3.8.13-cuda11.6.2-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.12.1-py3.9.12-ubuntu20.04`
	`docker pull spxiong/pytorch:1.12.1-py3.9.12-cuda11.3.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.12.1-py3.9.12-cuda11.3.1-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.12.1-py3.9.12-cuda11.6.2-ubuntu20.04`
	`docker pull spxiong/pytorch:1.12.1-py3.9.12-cuda11.6.2-devel-ubuntu20.04`

1.12.0

Image	Pull Command
	`docker pull spxiong/pytorch:1.12.0-py3.8.13-ubuntu20.04`
	`docker pull spxiong/pytorch:1.12.0-py3.8.13-cuda11.3.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.12.0-py3.8.13-cuda11.3.1-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.12.0-py3.8.13-cuda11.6.2-ubuntu20.04`
	`docker pull spxiong/pytorch:1.12.0-py3.8.13-cuda11.6.2-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.12.0-py3.9.12-ubuntu20.04`
	`docker pull spxiong/pytorch:1.12.0-py3.9.12-cuda11.3.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.12.0-py3.9.12-cuda11.3.1-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.12.0-py3.9.12-cuda11.6.2-ubuntu20.04`
	`docker pull spxiong/pytorch:1.12.0-py3.9.12-cuda11.6.2-devel-ubuntu20.04`

1.11.0

Image	Pull Command
	`docker pull spxiong/pytorch:1.11.0-py3.8.13-ubuntu20.04`
	`docker pull spxiong/pytorch:1.11.0-py3.8.13-cuda11.3.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.11.0-py3.8.13-cuda11.3.1-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.11.0-py3.9.12-ubuntu20.04`
	`docker pull spxiong/pytorch:1.11.0-py3.9.12-cuda11.3.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.11.0-py3.9.12-cuda11.3.1-devel-ubuntu20.04`

1.10.2

Image	Pull Command
	`docker pull spxiong/pytorch:1.10.2-py3.8.13-ubuntu20.04`
	`docker pull spxiong/pytorch:1.10.2-py3.8.13-cuda11.3.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.10.2-py3.8.13-cuda11.3.1-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.10.2-py3.9.12-ubuntu20.04`
	`docker pull spxiong/pytorch:1.10.2-py3.9.12-cuda11.3.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.10.2-py3.9.12-cuda11.3.1-devel-ubuntu20.04`

1.10.1

Image	Pull Command
	`docker pull spxiong/pytorch:1.10.1-py3.8.13-ubuntu20.04`
	`docker pull spxiong/pytorch:1.10.1-py3.8.13-cuda11.1.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.10.1-py3.8.13-cuda11.1.1-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.10.1-py3.9.12-ubuntu20.04`
	`docker pull spxiong/pytorch:1.10.1-py3.9.12-cuda11.1.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.10.1-py3.9.12-cuda11.1.1-devel-ubuntu20.04`

1.10.0

Image	Pull Command
	`docker pull spxiong/pytorch:1.10.0-py3.8.13-ubuntu20.04`
	`docker pull spxiong/pytorch:1.10.0-py3.8.13-cuda11.1.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.10.0-py3.8.13-cuda11.1.1-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.10.0-py3.9.12-ubuntu20.04`
	`docker pull spxiong/pytorch:1.10.0-py3.9.12-cuda11.1.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.10.0-py3.9.12-cuda11.1.1-devel-ubuntu20.04`

1.9.1

Image	Pull Command
	`docker pull spxiong/pytorch:1.9.1-py3.8.13-ubuntu20.04`
	`docker pull spxiong/pytorch:1.9.1-py3.8.13-cuda11.1.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.9.1-py3.8.13-cuda11.1.1-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.9.1-py3.9.12-ubuntu20.04`
	`docker pull spxiong/pytorch:1.9.1-py3.9.12-cuda11.1.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.9.1-py3.9.12-cuda11.1.1-devel-ubuntu20.04`

1.9.0

Image	Pull Command
	`docker pull spxiong/pytorch:1.9.0-py3.8.13-ubuntu20.04`
	`docker pull spxiong/pytorch:1.9.0-py3.8.13-cuda11.1.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.9.0-py3.8.13-cuda11.1.1-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.9.0-py3.9.12-ubuntu20.04`
	`docker pull spxiong/pytorch:1.9.0-py3.9.12-cuda11.1.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.9.0-py3.9.12-cuda11.1.1-devel-ubuntu20.04`

1.8.1

Image	Pull Command
	`docker pull spxiong/pytorch:1.8.1-py3.8.13-ubuntu20.04`
	`docker pull spxiong/pytorch:1.8.1-py3.8.13-cuda11.1.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.8.1-py3.8.13-cuda11.1.1-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.8.1-py3.9.12-ubuntu20.04`
	`docker pull spxiong/pytorch:1.8.1-py3.9.12-cuda11.1.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.8.1-py3.9.12-cuda11.1.1-devel-ubuntu20.04`

1.8.0

Image	Pull Command
	`docker pull spxiong/pytorch:1.8.0-py3.8.13-ubuntu20.04`
	`docker pull spxiong/pytorch:1.8.0-py3.8.13-cuda11.1.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.8.0-py3.8.13-cuda11.1.1-devel-ubuntu20.04`
	`docker pull spxiong/pytorch:1.8.0-py3.9.12-ubuntu20.04`
	`docker pull spxiong/pytorch:1.8.0-py3.9.12-cuda11.1.1-ubuntu20.04`
	`docker pull spxiong/pytorch:1.8.0-py3.9.12-cuda11.1.1-devel-ubuntu20.04`

1.7.1

Image	Pull Command
	`docker pull spxiong/pytorch:1.7.1-py3.8.13-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.1-py3.8.13-cuda10.1-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.1-py3.8.13-cuda10.1-devel-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.1-py3.8.13-cuda10.2-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.1-py3.8.13-cuda10.2-devel-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.1-py3.8.13-cuda11.0.3-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.1-py3.8.13-cuda11.0.3-devel-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.1-py3.9.12-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.1-py3.9.12-cuda10.1-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.1-py3.9.12-cuda10.1-devel-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.1-py3.9.12-cuda10.2-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.1-py3.9.12-cuda10.2-devel-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.1-py3.9.12-cuda11.0.3-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.1-py3.9.12-cuda11.0.3-devel-ubuntu18.04`

1.7.0

Image	Pull Command
	`docker pull spxiong/pytorch:1.7.0-py3.8.13-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.0-py3.8.13-cuda10.1-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.0-py3.8.13-cuda10.1-devel-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.0-py3.8.13-cuda10.2-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.0-py3.8.13-cuda10.2-devel-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.0-py3.8.13-cuda11.0.3-ubuntu18.04`
	`docker pull spxiong/pytorch:1.7.0-py3.8.13-cuda11.0.3-devel-ubuntu18.04`

1.6.0

Image	Pull Command
	`docker pull spxiong/pytorch:1.6.0-py3.8.13-ubuntu18.04`
	`docker pull spxiong/pytorch:1.6.0-py3.8.13-cuda10.1-ubuntu18.04`
	`docker pull spxiong/pytorch:1.6.0-py3.8.13-cuda10.1-devel-ubuntu18.04`
	`docker pull spxiong/pytorch:1.6.0-py3.8.13-cuda10.2-ubuntu18.04`
	`docker pull spxiong/pytorch:1.6.0-py3.8.13-cuda10.2-devel-ubuntu18.04`

1.5.1

Image	Pull Command
	`docker pull spxiong/pytorch:1.5.1-py3.8.13-ubuntu18.04`
	`docker pull spxiong/pytorch:1.5.1-py3.8.13-cuda10.1-ubuntu18.04`
	`docker pull spxiong/pytorch:1.5.1-py3.8.13-cuda10.1-devel-ubuntu18.04`
	`docker pull spxiong/pytorch:1.5.1-py3.8.13-cuda10.2-ubuntu18.04`
	`docker pull spxiong/pytorch:1.5.1-py3.8.13-cuda10.2-devel-ubuntu18.04`

1.5.0

Image	Pull Command
	`docker pull spxiong/pytorch:1.5.0-py3.8.13-ubuntu18.04`
	`docker pull spxiong/pytorch:1.5.0-py3.8.13-cuda10.1-ubuntu18.04`
	`docker pull spxiong/pytorch:1.5.0-py3.8.13-cuda10.1-devel-ubuntu18.04`
	`docker pull spxiong/pytorch:1.5.0-py3.8.13-cuda10.2-ubuntu18.04`
	`docker pull spxiong/pytorch:1.5.0-py3.8.13-cuda10.2-devel-ubuntu18.04`

1.4.0

Image	Pull Command
	`docker pull spxiong/pytorch:1.4.0-py3.8.13-ubuntu18.04`
	`docker pull spxiong/pytorch:1.4.0-py3.8.13-cuda10.1-ubuntu18.04`
	`docker pull spxiong/pytorch:1.4.0-py3.8.13-cuda10.1-devel-ubuntu18.04`

其他镜像

生成构建脚本

通过以下命令生成构建脚本（可用版本请参见可用版本）：

python generate_build_script.py --os <ubuntu 或 centos> --os-version <例如 20.04, 8> --python <例如 3.9.12> --pytorch <例如 1.9.1> --cuda <例如 11.1, cpu>

usage: generate_build_script.py [-h] --os OS --os-version OS_VERSION --python PYTHON --pytorch PYTORCH [--cuda CUDA] [--cuda-flavor CUDA_FLAVOR]

生成 docker 构建脚本。

可选参数:
  -h, --help            显示此帮助信息并退出
  --os OS               操作系统。
  --os-version OS_VERSION
                        操作系统版本。
  --python PYTHON       Python 版本。
  --pytorch PYTORCH     Pytorch 版本。
  --cuda CUDA           CUDA 版本，`cpu` 表示 CPU 版本。
  --cuda-flavor CUDA_FLAVOR
                        CUDA 风味，`runtime` 或 `devel`，默认是 None，表示使用基础镜像

构建 Pytorch Docker 镜像

scripts/build_xxx.sh

提交版本（可选）

如果你想使用 github actions 构建和发布特定版本，你可以 fork 这个仓库并提交一个 pull request。pull request 应该只包含 scripts/build_xxx.sh 和 generate_build_script.py 生成的 .github/workflows/docker_build_xxx.yml

Files

README_CN.md

Latest commit

History