+docker-compose.rocm.yml, +auto build rocm

This commit is contained in:
matatonic 2024-06-24 20:53:30 -04:00
parent 72c7b799b9
commit 756799529a
2 changed files with 89 additions and 1 deletions

View File

@ -69,7 +69,7 @@ jobs:
labels: version=${{ github.run_id }}
platforms: linux/amd64,linux/arm64
build-and-push-alt-image:
build-and-push-min-image:
runs-on: ubuntu-latest
permissions:
@ -129,3 +129,64 @@ jobs:
labels: version=${{ github.run_id }}
platforms: linux/amd64,linux/arm64
build-and-push-rocm-image:
runs-on: ubuntu-latest
permissions:
contents: read
packages: write
env:
# Set up environment variables for the job
USE_ROCM: 1
DOCKER_REGISTRY: ghcr.io
IMAGE_NAME: ${{ github.repository }}-rocm
TAG: ${{ github.sha }}
steps:
- name: Check out code
uses: actions/checkout@v4
- name: Set up Docker Buildx
uses: docker/setup-buildx-action@v2
with:
install: true
# Log in to the GitHub Container Registry only when not running on a pull request event
- name: Login to Docker Registry
uses: docker/login-action@v2
with:
registry: ${{ env.DOCKER_REGISTRY }}
username: ${{ github.actor }}
password: ${{ secrets.GITHUB_TOKEN }}
- name: Extract metadata (tags, labels) for Docker
id: meta
uses: docker/metadata-action@v4
with:
images: ${{ env.DOCKER_REGISTRY }}/${{ env.IMAGE_NAME }}
# Build and push the Docker image to GHCR for the main branch or specific tags
- name: Build and Push Docker Image
if: github.ref == 'refs/heads/main'
uses: docker/build-push-action@v4
with:
context: .
file: Dockerfile
push: true
tags: ${{ env.DOCKER_REGISTRY }}/${{ env.IMAGE_NAME }}:latest
labels: version=${{ github.run_id }}
platforms: linux/amd64,linux/arm64
# For tagged releases, build and push the Docker image with the corresponding tag
- name: Build and Push Docker Image (Tagged)
if: startsWith(github.ref, 'refs/tags/')
uses: docker/build-push-action@v4
with:
context: .
file: Dockerfile
push: true
tags: ${{ env.DOCKER_REGISTRY }}/${{ env.IMAGE_NAME }}:${{ github.ref_name }}
labels: version=${{ github.run_id }}
platforms: linux/amd64,linux/arm64

27
docker-compose.rocm.yml Normal file
View File

@ -0,0 +1,27 @@
services:
server:
build:
dockerfile: Dockerfile
image: ghcr.io/matatonic/openedai-speech-rocm
environment:
- USE_ROCM=1
env_file: speech.env
ports:
- "8000:8000"
volumes:
- ./voices:/app/voices
- ./config:/app/config
# To install as a service
restart: unless-stopped
# For AMD GPU (ROCm) Support
cap_add:
- SYS_PTRACE
devices:
- /dev/kfd
- /dev/dri
security_opt:
- seccomp=unconfined
group_add:
- video
- audio
ipc: host