File: Dockerfile

package info (click to toggle)
gpu-burn 0%2Bgit20240115%2Bds-5
  • links: PTS, VCS
  • area: non-free
  • in suites: sid, trixie
  • size: 188 kB
  • sloc: cpp: 1,380; makefile: 124
file content (19 lines) | stat: -rw-r--r-- 341 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
 
ARG CUDA_VERSION=11.8.0
ARG IMAGE_DISTRO=ubi8

FROM nvidia/cuda:${CUDA_VERSION}-devel-${IMAGE_DISTRO} AS builder

WORKDIR /build

COPY . /build/

RUN make

FROM nvidia/cuda:${CUDA_VERSION}-runtime-${IMAGE_DISTRO}

COPY --from=builder /build/gpu_burn /app/
COPY --from=builder /build/compare.ptx /app/

WORKDIR /app

CMD ["./gpu_burn", "60"]