feat: CUDA 12.6

doppeltilde · Oct 25, 2024 · 932d1b1 · 932d1b1
1 parent de7127a
commit 932d1b1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Dockerfile.cuda b/Dockerfile.cuda
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.6.2-runtime-ubuntu24.04
+FROM nvidia/cuda:12.6.2-base-ubuntu24.04
 WORKDIR /app
 COPY . /app
 RUN apt-get update && \

diff --git a/docker-compose.dev.cuda.yml b/docker-compose.dev.cuda.yml
@@ -1,10 +1,10 @@
 services:
-  cuda:
+  cuda_dev:
     build:
       context: .
       dockerfile: ./Dockerfile.cuda
     ports:
-      - "8000:8000"
+      - "8100:8000"
     volumes:
       - ./models:/root/.cache/huggingface/hub:rw
     environment: