Commit d594892
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -36,43 +36,4 @@ jobs: | |||
| 36 | 36 | push: true # push to registry | |
| 37 | 37 | pull: true # always fetch the latest base images | |
| 38 | 38 | platforms: linux/amd64,linux/arm64 # build for both amd64 and arm64 | |
| 39 | - tags: ghcr.io/abetlen/llama-cpp-python:latest | ||
| 40 | - | ||
| 41 | - docker-cuda: | ||
| 42 | - name: Build and push Docker CUDA image | ||
| 43 | - runs-on: ubuntu-latest | ||
| 44 | - steps: | ||
| 45 | - - name: Checkout | ||
| 46 | - uses: actions/checkout@v3 | ||
| 47 | - with: | ||
| 48 | - submodules: "true" | ||
| 49 | - | ||
| 50 | - - name: Setup CUDA 12.1 | ||
| 51 | - uses: Jimver/cuda-toolkit@v0.2.10 | ||
| 52 | - id: cuda-toolkit | ||
| 53 | - with: | ||
| 54 | - cuda: '12.1.0' | ||
| 55 | - method: network | ||
| 56 | - | ||
| 57 | - - name: Set up QEMU | ||
| 58 | - uses: docker/setup-qemu-action@v2 | ||
| 59 | - | ||
| 60 | - - name: Set up Docker Buildx | ||
| 61 | - uses: docker/setup-buildx-action@v2 | ||
| 62 | - | ||
| 63 | - - name: Login to GitHub Container Registry | ||
| 64 | - uses: docker/login-action@v2 | ||
| 65 | - with: | ||
| 66 | - registry: ghcr.io | ||
| 67 | - username: ${{ github.repository_owner }} | ||
| 68 | - password: ${{ secrets.GITHUB_TOKEN }} | ||
| 69 | - | ||
| 70 | - - name: Build and push | ||
| 71 | - uses: docker/build-push-action@v4 | ||
| 72 | - with: | ||
| 73 | - file: Dockerfile.cuda | ||
| 74 | - context: . | ||
| 75 | - push: true # push to registry | ||
| 76 | - pull: true # always fetch the latest base images | ||
| 77 | - platforms: linux/amd64,linux/arm64 # build for both amd64 and arm64 | ||
| 78 | - tags: ghcr.io/abetlen/llama-cpp-python-cuda:latest | ||
| 39 | + tags: ghcr.io/abetlen/llama-cpp-python:latest | ||
0 commit comments