Skip to content

Commit

Permalink
feature: 2.4.0_py3.10.15_ubuntu22.04
Browse files Browse the repository at this point in the history
  • Loading branch information
cnstark committed Oct 9, 2024
1 parent 62a1e2b commit 4112fd2
Show file tree
Hide file tree
Showing 7 changed files with 176 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
name: Build(2.4.0_py3.10.15_cuda12.1.0_devel_ubuntu22.04)

env:
BASE_IMAGE: "nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04"

PYTHON_VERSION: "3.10.15"

PYTORCH_VERSION: "2.4.0"
PYTORCH_VERSION_SUFFIX: "+cu121"
TORCHVISION_VERSION: "0.19.0"
TORCHVISION_VERSION_SUFFIX: "+cu121"
TORCHAUDIO_VERSION: "2.4.0"
TORCHAUDIO_VERSION_SUFFIX: "+cu121"
PYTORCH_DOWNLOAD_URL: "https://download.pytorch.org/whl/cu121/torch_stable.html"

IMAGE_TAG: "2.4.0-py3.10.15-cuda12.1.0-devel-ubuntu22.04"

on:
push:
branches:
- main
paths:
- 'docker/ubuntu/**'
- '.github/workflows/docker_build_2.4.0_py3.10.15_cuda12.1.0_devel_ubuntu22.04.yml'

jobs:
build:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v2

- name: Login DockerHub
run: docker login --username=${{ secrets.DOCKER_USERNAME }} --password=${{ secrets.DOCKER_PASSWORD }}

- name: Build docker image
run: docker/ubuntu/build.sh

- name: Push docker image
run: docker push cnstark/pytorch:${IMAGE_TAG}
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
name: Build(2.4.0_py3.10.15_cuda12.1.0_ubuntu22.04)

env:
BASE_IMAGE: "ubuntu:22.04"

PYTHON_VERSION: "3.10.15"

PYTORCH_VERSION: "2.4.0"
PYTORCH_VERSION_SUFFIX: "+cu121"
TORCHVISION_VERSION: "0.19.0"
TORCHVISION_VERSION_SUFFIX: "+cu121"
TORCHAUDIO_VERSION: "2.4.0"
TORCHAUDIO_VERSION_SUFFIX: "+cu121"
PYTORCH_DOWNLOAD_URL: "https://download.pytorch.org/whl/cu121/torch_stable.html"

IMAGE_TAG: "2.4.0-py3.10.15-cuda12.1.0-ubuntu22.04"

on:
push:
branches:
- main
paths:
- 'docker/ubuntu/**'
- '.github/workflows/docker_build_2.4.0_py3.10.15_cuda12.1.0_ubuntu22.04.yml'

jobs:
build:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v2

- name: Login DockerHub
run: docker login --username=${{ secrets.DOCKER_USERNAME }} --password=${{ secrets.DOCKER_PASSWORD }}

- name: Build docker image
run: docker/ubuntu/build.sh

- name: Push docker image
run: docker push cnstark/pytorch:${IMAGE_TAG}
39 changes: 39 additions & 0 deletions .github/workflows/docker_build_2.4.0_py3.10.15_ubuntu22.04.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
name: Build(2.4.0_py3.10.15_ubuntu22.04)

env:
BASE_IMAGE: "ubuntu:22.04"

PYTHON_VERSION: "3.10.15"

PYTORCH_VERSION: "2.4.0"
PYTORCH_VERSION_SUFFIX: "+cpu"
TORCHVISION_VERSION: "0.19.0"
TORCHVISION_VERSION_SUFFIX: "+cpu"
TORCHAUDIO_VERSION: "2.4.0"
TORCHAUDIO_VERSION_SUFFIX: "+cpu"
PYTORCH_DOWNLOAD_URL: "https://download.pytorch.org/whl/cpu/torch_stable.html"

IMAGE_TAG: "2.4.0-py3.10.15-ubuntu22.04"

on:
push:
branches:
- main
paths:
- 'docker/ubuntu/**'
- '.github/workflows/docker_build_2.4.0_py3.10.15_ubuntu22.04.yml'

jobs:
build:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v2

- name: Login DockerHub
run: docker login --username=${{ secrets.DOCKER_USERNAME }} --password=${{ secrets.DOCKER_PASSWORD }}

- name: Build docker image
run: docker/ubuntu/build.sh

- name: Push docker image
run: docker push cnstark/pytorch:${IMAGE_TAG}
8 changes: 8 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -119,6 +119,14 @@ Note: `/path/to/project` and `/path/to/dataset` is your **own** project path and
<summary>More Images</summary>
</details> -->

### 2.4.0

| Image | Pull Command |
| -------------| -------------|
| ![pytorch2.4.0] ![python3.10.15] ![cuda12.1] ![ubuntu22.04] [![](https://img.shields.io/docker/image-size/cnstark/pytorch/2.4.0-py3.10.15-cuda12.1.0-ubuntu22.04)][DockerHub] | `docker pull cnstark/pytorch:2.4.0-py3.10.15-cuda12.1.0-ubuntu22.04` |
| ![pytorch2.4.0] ![python3.10.15] ![cuda12.1-devel] ![ubuntu22.04] [![](https://img.shields.io/docker/image-size/cnstark/pytorch/2.4.0-py3.10.15-cuda12.1.0-devel-ubuntu22.04)][DockerHub] | `docker pull cnstark/pytorch:2.4.0-py3.10.15-cuda12.1.0-devel-ubuntu22.04` |
| ![pytorch2.4.0] ![python3.10.15] ![cpu] ![ubuntu22.04] [![](https://img.shields.io/docker/image-size/cnstark/pytorch/2.4.0-py3.10.15-ubuntu22.04)][DockerHub] | `docker pull cnstark/pytorch:2.4.0-py3.10.15-ubuntu22.04` |

### 2.3.1

| Image | Pull Command |
Expand Down
17 changes: 17 additions & 0 deletions scripts/build_2.4.0_py3.10.15_cuda12.1.0_devel_ubuntu22.04.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
#!/bin/sh

export BASE_IMAGE=nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04

export PYTHON_VERSION=3.10.15

export PYTORCH_VERSION=2.4.0
export PYTORCH_VERSION_SUFFIX=+cu121
export TORCHVISION_VERSION=0.19.0
export TORCHVISION_VERSION_SUFFIX=+cu121
export TORCHAUDIO_VERSION=2.4.0
export TORCHAUDIO_VERSION_SUFFIX=+cu121
export PYTORCH_DOWNLOAD_URL=https://download.pytorch.org/whl/cu121/torch_stable.html

export IMAGE_TAG=2.4.0-py3.10.15-cuda12.1.0-devel-ubuntu22.04

./docker/ubuntu/build.sh
17 changes: 17 additions & 0 deletions scripts/build_2.4.0_py3.10.15_cuda12.1.0_ubuntu22.04.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
#!/bin/sh

export BASE_IMAGE=ubuntu:22.04

export PYTHON_VERSION=3.10.15

export PYTORCH_VERSION=2.4.0
export PYTORCH_VERSION_SUFFIX=+cu121
export TORCHVISION_VERSION=0.19.0
export TORCHVISION_VERSION_SUFFIX=+cu121
export TORCHAUDIO_VERSION=2.4.0
export TORCHAUDIO_VERSION_SUFFIX=+cu121
export PYTORCH_DOWNLOAD_URL=https://download.pytorch.org/whl/cu121/torch_stable.html

export IMAGE_TAG=2.4.0-py3.10.15-cuda12.1.0-ubuntu22.04

./docker/ubuntu/build.sh
17 changes: 17 additions & 0 deletions scripts/build_2.4.0_py3.10.15_ubuntu22.04.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
#!/bin/sh

export BASE_IMAGE=ubuntu:22.04

export PYTHON_VERSION=3.10.15

export PYTORCH_VERSION=2.4.0
export PYTORCH_VERSION_SUFFIX=+cpu
export TORCHVISION_VERSION=0.19.0
export TORCHVISION_VERSION_SUFFIX=+cpu
export TORCHAUDIO_VERSION=2.4.0
export TORCHAUDIO_VERSION_SUFFIX=+cpu
export PYTORCH_DOWNLOAD_URL=https://download.pytorch.org/whl/cpu/torch_stable.html

export IMAGE_TAG=2.4.0-py3.10.15-ubuntu22.04

./docker/ubuntu/build.sh

0 comments on commit 4112fd2

Please sign in to comment.