mirror of
https://github.com/pytorch/pytorch.git
synced 2025-11-04 16:04:58 +08:00
b29d77b54f
Fixes https://github.com/pytorch/pytorch/issues/125094 Please note: Docker CUDa 12.4 failure is existing issue, related to docker image not being available on gitlab: ``` docker.io/nvidia/cuda:12.4.0-cudnn8-devel-ubuntu22.04: docker.io/nvidia/cuda:12.4.0-cudnn8-devel-ubuntu22.04: not found ``` https://github.com/pytorch/pytorch/actions/runs/8974959068/job/24648540236?pr=125617 Here is the reference issue: https://gitlab.com/nvidia/container-images/cuda/-/issues/225 Tracked on our side: https://github.com/pytorch/builder/issues/1811 Pull Request resolved: https://github.com/pytorch/pytorch/pull/125617 Approved by: https://github.com/huydhn, https://github.com/malfet
55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
|
|
"""Generates a matrix for docker releases through github actions
|
|
|
|
Will output a condensed version of the matrix. Will include fllowing:
|
|
* CUDA version short
|
|
* CUDA full version
|
|
* CUDNN version short
|
|
* Image type either runtime or devel
|
|
* Platform linux/arm64,linux/amd64
|
|
|
|
"""
|
|
|
|
import json
|
|
from typing import Dict, List
|
|
|
|
import generate_binary_build_matrix
|
|
|
|
DOCKER_IMAGE_TYPES = ["runtime", "devel"]
|
|
|
|
|
|
def generate_docker_matrix() -> Dict[str, List[Dict[str, str]]]:
|
|
ret: List[Dict[str, str]] = []
|
|
# CUDA amd64 Docker images are available as both runtime and devel while
|
|
# CPU arm64 image is only available as runtime.
|
|
for cuda, version in generate_binary_build_matrix.CUDA_ARCHES_FULL_VERSION.items():
|
|
for image in DOCKER_IMAGE_TYPES:
|
|
ret.append(
|
|
{
|
|
"cuda": cuda,
|
|
"cuda_full_version": version,
|
|
"cudnn_version": generate_binary_build_matrix.CUDA_ARCHES_CUDNN_VERSION[
|
|
cuda
|
|
],
|
|
"image_type": image,
|
|
"platform": "linux/amd64",
|
|
}
|
|
)
|
|
ret.append(
|
|
{
|
|
"cuda": "cpu",
|
|
"cuda_full_version": "",
|
|
"cudnn_version": "",
|
|
"image_type": "runtime",
|
|
"platform": "linux/arm64",
|
|
}
|
|
)
|
|
|
|
return {"include": ret}
|
|
|
|
|
|
if __name__ == "__main__":
|
|
build_matrix = generate_docker_matrix()
|
|
print(json.dumps(build_matrix))
|