Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

docker: CUDA with runtime (#17977)

This commit is contained in:
Jirka Borovec 2023-07-03 17:39:09 +02:00 committed by GitHub
parent 8de56eaac7
commit 4265c11e8c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
.github/checkgroup.yml vendored
View File

@ -141,14 +141,14 @@ subprojects:
- "!**/*.md"
checks:
- "build-cuda (3.9, 1.11, 11.3.1)"
- "build-cuda (3.9, 1.12, 11.6.1)"
- "build-cuda (3.9, 1.13, 11.7.1)"
- "build-cuda (3.10, 2.0, 11.7.1)"
- "build-cuda (3.9, 1.12, 11.7.1)"
- "build-cuda (3.9, 1.13, 12.0.1)"
- "build-cuda (3.10, 2.0, 12.0.1)"
#- "build-NGC"
- "build-pl (3.9, 1.11, 11.3.1)"
- "build-pl (3.9, 1.12, 11.6.1)"
- "build-pl (3.9, 1.13, 11.7.1)"
- "build-pl (3.10, 2.0, 11.7.1)"
- "build-pl (3.9, 1.12, 11.7.1)"
- "build-pl (3.9, 1.13, 12.0.1)"
- "build-pl (3.10, 2.0, 12.0.1)"
# SECTIONS: lightning_data

12
.github/workflows/ci-dockers.yml vendored
View File

@ -44,9 +44,9 @@ jobs:
# We only release one docker image per PyTorch version.
# The matrix here is the same as the one in release-docker.yml.
- {python_version: "3.9", pytorch_version: "1.11", cuda_version: "11.3.1"}
- {python_version: "3.9", pytorch_version: "1.12", cuda_version: "11.6.1"}
- {python_version: "3.9", pytorch_version: "1.13", cuda_version: "11.7.1"}
- {python_version: "3.10", pytorch_version: "2.0", cuda_version: "11.7.1"}
- {python_version: "3.9", pytorch_version: "1.12", cuda_version: "11.7.1"}
- {python_version: "3.9", pytorch_version: "1.13", cuda_version: "12.0.1"}
- {python_version: "3.10", pytorch_version: "2.0", cuda_version: "12.0.1"}
steps:
- uses: actions/checkout@v3
with:
@ -72,9 +72,9 @@ jobs:
# These are the base images for PL release docker images,
# so include at least all of the combinations in release-dockers.yml.
- {python_version: "3.9", pytorch_version: "1.11", cuda_version: "11.3.1"}
- {python_version: "3.9", pytorch_version: "1.12", cuda_version: "11.6.1"}
- {python_version: "3.9", pytorch_version: "1.13", cuda_version: "11.7.1"}
- {python_version: "3.10", pytorch_version: "2.0", cuda_version: "11.7.1"}
- {python_version: "3.9", pytorch_version: "1.12", cuda_version: "11.7.1"}
- {python_version: "3.9", pytorch_version: "1.13", cuda_version: "12.0.1"}
- {python_version: "3.10", pytorch_version: "2.0", cuda_version: "12.0.1"}
steps:
- uses: actions/checkout@v3
- uses: docker/setup-buildx-action@v2

4
.github/workflows/release-docker.yml vendored
View File

@ -17,8 +17,8 @@ jobs:
include:
# We only release one docker image per PyTorch version.
- {python_version: "3.9", pytorch_version: "1.11", cuda_version: "11.3.1"}
- {python_version: "3.9", pytorch_version: "1.12", cuda_version: "11.6.1"}
- {python_version: "3.9", pytorch_version: "1.13", cuda_version: "11.7.1"}
- {python_version: "3.9", pytorch_version: "1.12", cuda_version: "11.7.1"}
- {python_version: "3.9", pytorch_version: "1.13", cuda_version: "12.0.1"}
steps:
- name: Checkout
uses: actions/checkout@v3

2
dockers/base-cuda/Dockerfile
View File

@ -16,7 +16,7 @@ ARG UBUNTU_VERSION=20.04
ARG CUDA_VERSION=11.7.1
FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}
FROM nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu${UBUNTU_VERSION}
ARG PYTHON_VERSION=3.10
ARG PYTORCH_VERSION=2.0