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

Update building docker images (#12837) 

Co-authored-by: Akihiro Nitta <akihiro@pytorchlightning.ai>
This commit is contained in:
Akihiro Nitta 2022-04-22 07:10:42 +09:00 committed by lexierule
parent 65c3cece16
commit 55503b3138
3 changed files with 14 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
.github/workflows/ci_dockers.yml vendored
View File

@ -73,11 +73,14 @@ jobs:
strategy: strategy:
fail-fast: false fail-fast: false
matrix: matrix:
python_version: ["3.7", "3.9"]
pytorch_version: ["1.8", "1.11"]
include: include:
# the config used in '.azure-pipelines/gpu-tests.yml' # the config used in '.azure-pipelines/gpu-tests.yml'
- {python_version: "3.9", pytorch_version: "1.10"} - {python_version: "3.7", pytorch_version: "1.10", cuda_version: "11.1"}
- {python_version: "3.7", pytorch_version: "1.11", cuda_version: "11.3.1"}
# latest (used in Tutorials)
- {python_version: "3.8", pytorch_version: "1.8", cuda_version: "11.1"}
- {python_version: "3.9", pytorch_version: "1.10", cuda_version: "11.1"}
- {python_version: "3.9", pytorch_version: "1.11", cuda_version: "11.3.1"}
steps: steps:
- name: Checkout - name: Checkout
uses: actions/checkout@v2 uses: actions/checkout@v2
@ -88,6 +91,7 @@ jobs:
build-args: | build-args: |
PYTHON_VERSION=${{ matrix.python_version }} PYTHON_VERSION=${{ matrix.python_version }}
PYTORCH_VERSION=${{ matrix.pytorch_version }} PYTORCH_VERSION=${{ matrix.pytorch_version }}
CUDA_VERSION=${{ matrix.cuda_version }}
file: dockers/base-cuda/Dockerfile file: dockers/base-cuda/Dockerfile
push: false push: false
timeout-minutes: 75 timeout-minutes: 75

11
.github/workflows/events-nightly.yml vendored
View File

@ -115,12 +115,12 @@ jobs:
matrix: matrix:
include: include:
# the config used in '.azure-pipelines/gpu-tests.yml' # the config used in '.azure-pipelines/gpu-tests.yml'
- {python_version: "3.7", pytorch_version: "1.8"} - {python_version: "3.7", pytorch_version: "1.10", cuda_version: "11.1"}
- {python_version: "3.7", pytorch_version: "1.10"} - {python_version: "3.7", pytorch_version: "1.11", cuda_version: "11.3.1"}
# latest (used in Tutorials) # latest (used in Tutorials)
- {python_version: "3.8", pytorch_version: "1.8"} - {python_version: "3.8", pytorch_version: "1.8", cuda_version: "11.1"}
- {python_version: "3.9", pytorch_version: "1.10"} - {python_version: "3.9", pytorch_version: "1.10", cuda_version: "11.1"}
- {python_version: "3.9", pytorch_version: "1.11"} - {python_version: "3.9", pytorch_version: "1.11", cuda_version: "11.3.1"}
steps: steps:
- name: Checkout - name: Checkout
@ -140,6 +140,7 @@ jobs:
build-args: | build-args: |
PYTHON_VERSION=${{ matrix.python_version }} PYTHON_VERSION=${{ matrix.python_version }}
PYTORCH_VERSION=${{ matrix.pytorch_version }} PYTORCH_VERSION=${{ matrix.pytorch_version }}
CUDA_VERSION=${{ matrix.cuda_version }}
file: dockers/base-cuda/Dockerfile file: dockers/base-cuda/Dockerfile
push: ${{ env.PUSH_TO_HUB }} push: ${{ env.PUSH_TO_HUB }}
tags: pytorchlightning/pytorch_lightning:base-cuda-py${{ matrix.python_version }}-torch${{ matrix.pytorch_version }} tags: pytorchlightning/pytorch_lightning:base-cuda-py${{ matrix.python_version }}-torch${{ matrix.pytorch_version }}

2
dockers/base-cuda/Dockerfile
View File

@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
ARG CUDA_VERSION=11.1 ARG CUDA_VERSION=11.3.1
FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04 FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04