vllm - 💡(How to fix) Fix [Feature]: upstream nightly rocm docker [4 comments, 3 participants]

# Dockerfile for ROCm
FROM ubuntu:20.04

# Install dependencies
RUN apt-get update && apt-get install -y \
    rocminfo \
    rocm-dev \
    hip-sdk

# Set environment variables
ENV ROCM_HOME=/opt/rocm
ENV PATH=$ROCM_HOME/bin:$PATH
ENV LD_LIBRARY_PATH=$ROCM_HOME/lib:$LD_LIBRARY_PATH

# Verify ROCm installation
RUN rocminfo

# GitHub Actions workflow file
name: Build and Push ROCm Docker Image

on:
  schedule:
    - cron: 0 0 * * *

jobs:
  build-and-push:
    runs-on: ubuntu-latest
    steps:
      - name: Checkout code
        uses: actions/checkout@v2
      - name: Login to Docker Hub
        uses: docker/login-action@v1
        with:
          username: ${{ secrets.DOCKER_USERNAME }}
          password: ${{ secrets.DOCKER_PASSWORD }}
      - name: Build and push Docker image
        run: |
          docker build -t my-rocm-image .
          docker tag my-rocm-image ${{ secrets.DOCKER_USERNAME }}/my-rocm-image:latest
          docker push ${{ secrets.DOCKER_USERNAME }}/my-rocm-image:latest