File tree Expand file tree Collapse file tree 1 file changed +38
-0
lines changed
Filter options
Expand file tree Collapse file tree 1 file changed +38
-0
lines changed
Original file line number Diff line number Diff line change 3737 pull : true # always fetch the latest base images
3838 platforms : linux/amd64,linux/arm64 # build for both amd64 and arm64
3939 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+
56+ - name : Set up QEMU
57+ uses : docker/setup-qemu-action@v2
58+
59+ - name : Set up Docker Buildx
60+ uses : docker/setup-buildx-action@v2
61+
62+ - name : Login to GitHub Container Registry
63+ uses : docker/login-action@v2
64+ with :
65+ registry : ghcr.io
66+ username : ${{ github.repository_owner }}
67+ password : ${{ secrets.GITHUB_TOKEN }}
68+
69+ - name : Build and push
70+ uses : docker/build-push-action@v4
71+ with :
72+ file : Dockerfile.cuda
73+ context : .
74+ push : true # push to registry
75+ pull : true # always fetch the latest base images
76+ platforms : linux/amd64,linux/arm64 # build for both amd64 and arm64
77+ tags : ghcr.io/abetlen/llama-cpp-python-cuda:latest
You can’t perform that action at this time.
0 commit comments