From 5b4ce08d99220a839ac76e336773d4de7208babf Mon Sep 17 00:00:00 2001
From: Googlefan <google@neody.dev>
Date: Wed, 11 Sep 2024 05:35:56 +0000
Subject: [PATCH] fix: docker script

---
 docker/cuda.Dockerfile | 4 ++--
 docker/run.sh          | 5 ++++-
 2 files changed, 6 insertions(+), 3 deletions(-)
 mode change 100644 => 100755 docker/run.sh

diff --git a/docker/cuda.Dockerfile b/docker/cuda.Dockerfile
index dd5215e..1b1b637 100644
--- a/docker/cuda.Dockerfile
+++ b/docker/cuda.Dockerfile
@@ -2,9 +2,9 @@ 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
+FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04
 WORKDIR /work
 COPY --from=builder /work/target/release/sbv2_api /work/main
 COPY --from=builder /work/target/release/*.so /work
+ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/work
 CMD ["/work/main"]
\ No newline at end of file
diff --git a/docker/run.sh b/docker/run.sh
old mode 100644
new mode 100755
index 7b32775..38e6afa
--- a/docker/run.sh
+++ b/docker/run.sh
@@ -1 +1,4 @@
-docker run -it --rm -p 3000:3000 --name sbv2 -v ./models:/work/models --env-file .env sbv2
\ No newline at end of file
+docker run -it --rm -p 3000:3000 --name sbv2 \
+-v ./models:/work/models --env-file .env \
+--gpus all \
+ghcr.io/tuna2134/sbv2-api:cuda
\ No newline at end of file