diff --git a/docker/cuda.Dockerfile b/docker/cuda.Dockerfile new file mode 100644 index 0000000..dd5215e --- /dev/null +++ b/docker/cuda.Dockerfile @@ -0,0 +1,10 @@ +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"] \ No newline at end of file