Spaces:
Sleeping
Sleeping
add CUDA11.4.Dockerfile
Browse files- CUDA11.4.Dockerfile +25 -0
CUDA11.4.Dockerfile
ADDED
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
FROM nvidia/cuda:11.4.3-devel-ubuntu20.04 as chef
|
2 |
+
RUN apt-get update && apt-get install -y curl
|
3 |
+
RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- --default-toolchain stable -y
|
4 |
+
ENV PATH=/root/.cargo/bin:$PATH
|
5 |
+
RUN cargo install cargo-chef
|
6 |
+
WORKDIR /app
|
7 |
+
|
8 |
+
FROM chef as planner
|
9 |
+
COPY . .
|
10 |
+
RUN cargo chef prepare --recipe-path recipe.json
|
11 |
+
|
12 |
+
FROM chef as builder
|
13 |
+
ENV DEBIAN_FRONTEND=noninteractive
|
14 |
+
RUN apt-get update && apt-get install -y cmake g++ libclang-dev libssl-dev pkg-config python3-dev
|
15 |
+
COPY --from=planner /app/recipe.json recipe.json
|
16 |
+
RUN cargo chef cook --release --recipe-path recipe.json
|
17 |
+
COPY . .
|
18 |
+
RUN cargo build --release
|
19 |
+
|
20 |
+
FROM nvidia/cuda:11.4.3-runtime-ubuntu20.04 as runtime
|
21 |
+
|
22 |
+
RUN apt-get update && apt-get install -y ca-certificates && rm -rf /var/lib/apt/lists/*
|
23 |
+
WORKDIR /app
|
24 |
+
COPY --from=builder /app/target/release/polyhedron .
|
25 |
+
ENTRYPOINT ["./polyhedron"]
|