Skip to content

Commit

Permalink
add cuda
Browse files Browse the repository at this point in the history
  • Loading branch information
kaylorchen committed Nov 18, 2024
1 parent 81802c7 commit 207e8aa
Show file tree
Hide file tree
Showing 3 changed files with 89 additions and 0 deletions.
39 changes: 39 additions & 0 deletions .github/workflows/build-l4t-r36.3_cuda.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
name: Build and push l4t r36.3 cuda Docker image

on:
push:
paths:
- 'l4t-base/Dockerfile_r36.3.cuda'
- 'l4t-base/packages_cuda.list'
pull_request:
paths:
- 'l4t-base/packages_cuda.list'
- 'l4t-base/Dockerfile_r36.3.cuda'

jobs:
build-and-push:
runs-on: ubuntu-latest
steps:
- name: Checkout code
uses: actions/checkout@v2

- name: Set up Docker Buildx
uses: docker/setup-buildx-action@v1

- name: Login to DockerHub
uses: docker/login-action@v1
with:
username: ${{ secrets.DOCKERHUB_USERNAME }}
password: ${{ secrets.DOCKERHUB_TOKEN }}

- name: Build and push
uses: docker/build-push-action@v2
with:
context: .
file: l4t-base/Dockerfile_r36.3.cuda
push: true
tags: kaylor/l4t-base:r36.3.0_cuda
platforms: linux/arm64

- name: Logout of DockerHub
run: docker logout
16 changes: 16 additions & 0 deletions l4t-base/Dockerfile_r36.3_cuda
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
FROM kaylor/l4t-base:r36.3.0_base

ENV DEBIAN_FRONTEND=noninteractive
ENV TZ=Asia/Shanghai
# 维护者
LABEL maintainer="[email protected]"

COPY l4t-base/packages.list packages.list
COPY l4t-base/packages_cuda.list packages_cuda.list

RUN mv /etc/apt/sources.list.d/nvidia-l4t-apt-source.list /etc/apt/sources.list.d/nvidia-l4t-apt-source.list.save && \
apt-get update && \
apt install -y --no-install-recommends --allow-downgrades $(sed ':label;N;s/\n/ /;b label' packages_cuda.list) && \
apt-get clean && \
rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*

34 changes: 34 additions & 0 deletions l4t-base/packages_cuda.list
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
cuda-cccl-12-2=12.2.140-1
cuda-command-line-tools-12-2=12.2.12-1
cuda-compat-12-2=12.2.34086590-1
cuda-compiler-12-2=12.2.12-1
cuda-crt-12-2=12.2.140-1
cuda-cudart-12-2=12.2.140-1
cuda-cudart-dev-12-2=12.2.140-1
cuda-cuobjdump-12-2=12.2.140-1
cuda-cupti-12-2=12.2.142-1
cuda-cupti-dev-12-2=12.2.142-1
cuda-cuxxfilt-12-2=12.2.140-1
cuda-documentation-12-2=12.2.140-1
cuda-driver-dev-12-2=12.2.140-1
cuda-gdb-12-2=12.2.140-1
cuda-libraries-12-2=12.2.12-1
cuda-libraries-dev-12-2=12.2.12-1
cuda-nsight-compute-12-2=12.2.12-1
cuda-nvcc-12-2=12.2.140-1
cuda-nvdisasm-12-2=12.2.140-1
cuda-nvml-dev-12-2=12.2.140-1
cuda-nvprune-12-2=12.2.140-1
cuda-nvrtc-12-2=12.2.140-1
cuda-nvrtc-dev-12-2=12.2.140-1
cuda-nvtx-12-2=12.2.140-1
cuda-nvvm-12-2=12.2.140-1
cuda-profiler-api-12-2=12.2.140-1
cuda-runtime-12-2=12.2.12-1
cuda-sanitizer-12-2=12.2.140-1
cuda-toolkit-12-2=12.2.12-1
cuda-toolkit-12-2-config-common=12.2.140-1
cuda-toolkit-12-config-common=12.2.140-1
cuda-toolkit-config-common=12.2.140-1
cuda-tools-12-2=12.2.12-1
cuda-visual-tools-12-2=12.2.12-1

0 comments on commit 207e8aa

Please sign in to comment.