File tree Expand file tree Collapse file tree 1 file changed +1
-40
lines changed Expand file tree Collapse file tree 1 file changed +1
-40
lines changed Original file line number Diff line number Diff line change 3636 push : true # push to registry
3737 pull : true # always fetch the latest base images
3838 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
You can’t perform that action at this time.
0 commit comments