Skip to content

compatible with torch 2.3 (#52) #36

compatible with torch 2.3 (#52)

compatible with torch 2.3 (#52) #36

name: Build and Publish Docker
on:
push:
branches:
- main
- cuda12
jobs:
docker:
runs-on: self-hosted
steps:
-
name: Checkout
uses: actions/checkout@v3
-
name: Set up QEMU
uses: docker/setup-qemu-action@v2
-
name: Set up Docker Buildx
uses: docker/setup-buildx-action@v2
-
name: Login to DockerHub
uses: docker/login-action@v2
with:
username: ${{ secrets.DOCKERHUB_USERNAME }}
password: ${{ secrets.DOCKERHUB_TOKEN }}
-
name: Build and push cu12.4 with rdma
uses: docker/build-push-action@v3
with:
context: ./docker/rdma/
push: true
tags: dptechnology/unicore:latest-pytorch2.3.0-cuda12.4-rdma