mirror of
https://github.com/neodyland/sbv2-api.git
synced 2026-01-04 05:22:56 +00:00
10 lines
292 B
Docker
10 lines
292 B
Docker
FROM rust AS builder
|
|
WORKDIR /work
|
|
COPY . .
|
|
RUN cargo build -r --bin sbv2_api -F cuda,cuda_tf32
|
|
|
|
FROM nvidia/cuda:12.6.1-cudnn-runtime-ubuntu24.04
|
|
WORKDIR /work
|
|
COPY --from=builder /work/target/release/sbv2_api /work/main
|
|
COPY --from=builder /work/target/release/*.so /work
|
|
CMD ["/work/main"] |