MVPilgrim commited on
Commit
0d43e84
·
1 Parent(s): ed020a6
Files changed (2) hide show
  1. Dockerfile +4 -4
  2. startup.sh +4 -1
Dockerfile CHANGED
@@ -1,5 +1,5 @@
1
  # Start with NVIDIA CUDA 12.6 base image
2
- FROM nvidia/cuda:12.6.0-base-ubuntu22.04 AS base
3
 
4
  ENV DEBIAN_FRONTEND=noninteractive
5
  ENV DEBCONF_NOWARNINGS="yes"
@@ -17,9 +17,9 @@ RUN apt-get update && apt-get install -y \
17
  git \
18
  build-essential \
19
  pkg-config \
20
- cuda-toolkit-12-6 \
21
- cuda-nvcc-12-6 \
22
- libcublas-12-6 \
23
  libcudnn8 \
24
  && apt-get clean \
25
  && rm -rf /var/lib/apt/lists/*
 
1
  # Start with NVIDIA CUDA 12.6 base image
2
+ FROM nvidia/cuda:12.2.0-base-ubuntu22.04 AS base
3
 
4
  ENV DEBIAN_FRONTEND=noninteractive
5
  ENV DEBCONF_NOWARNINGS="yes"
 
17
  git \
18
  build-essential \
19
  pkg-config \
20
+ cuda-toolkit-12-2 \
21
+ cuda-nvcc-12-2 \
22
+ libcublas-12-2 \
23
  libcudnn8 \
24
  && apt-get clean \
25
  && rm -rf /var/lib/apt/lists/*
startup.sh CHANGED
@@ -7,7 +7,7 @@
7
  #####################################
8
  exec &> /app/startup.log
9
 
10
- echo "#### pip3 list"; pip3 list
11
 
12
  stdbuf -o0 echo "#### startup.sh entered."
13
  #stdbuf -o0 echo "### ps -ef 1"; ps -ef
@@ -49,6 +49,9 @@ export AUTHENTICATION_ANONYMOUS_ACCESS_ENABLED=true \
49
  TRANSFORMERS_INFERENCE_API=http://127.0.0.1:8081 \
50
  LOG_LEVEL=warning \
51
  MODULES_CLIENT_TIMEOUT=600s
 
 
 
52
  /app/weaviate/weaviate --host 127.0.0.1 --port 8080 --scheme http --write-timeout 600s &
53
  stdbuf -o0 echo "### After Weaviate DB start. RC=$?"
54
 
 
7
  #####################################
8
  exec &> /app/startup.log
9
 
10
+ #echo "#### pip3 list"; pip3 list
11
 
12
  stdbuf -o0 echo "#### startup.sh entered."
13
  #stdbuf -o0 echo "### ps -ef 1"; ps -ef
 
49
  TRANSFORMERS_INFERENCE_API=http://127.0.0.1:8081 \
50
  LOG_LEVEL=warning \
51
  MODULES_CLIENT_TIMEOUT=600s
52
+
53
+ #echo "### Before sleep 10 before starting Weaviate."
54
+ sleep 10
55
  /app/weaviate/weaviate --host 127.0.0.1 --port 8080 --scheme http --write-timeout 600s &
56
  stdbuf -o0 echo "### After Weaviate DB start. RC=$?"
57