Reduce docker build context size

ludoplex · Feb 13, 2023 · 6ac8d0b · 6ac8d0b
1 parent 60674db
commit 6ac8d0b
Show file tree

Hide file tree

Showing 10 changed files with 60 additions and 354 deletions.
diff --git a/.dockerignore b/.dockerignore
diff --git a/api.dockerignore b/api.dockerignore
@@ -0,0 +1,8 @@
+.hf_cache/
+.idea
+*.md
+.git*
+.enditorconfig
+models
+converter
+tests
diff --git a/docker-compose.yaml b/docker-compose.yaml
@@ -3,7 +3,7 @@ services:
   triton:
     build:
       context: .
-      dockerfile: Dockerfile
+      dockerfile: triton.Dockerfile
     command: bash -c "CUDA_VISIBLE_DEVICES=${GPUS} mpirun -n 1 --allow-run-as-root /opt/tritonserver/bin/tritonserver --model-repository=/model"
     shm_size: '2gb'
     volumes:
@@ -27,7 +27,7 @@ services:
     # For local build
     build:
       context: .
-      dockerfile: copilot_proxy/Dockerfile
+      dockerfile: proxy.Dockerfile
     env_file:
       # Automatically created via ./setup.sh
       - .env

diff --git a/copilot_proxy/Dockerfile → proxy.Dockerfile b/copilot_proxy/Dockerfile → proxy.Dockerfile
diff --git a/tests/python_backend/docker-compose-with-gpus.yaml b/tests/python_backend/docker-compose-with-gpus.yaml
@@ -3,7 +3,7 @@ services:
   triton:
     build:
       context: ../../
-      dockerfile: Dockerfile
+      dockerfile: ../../triton.Dockerfile
     command: bash -c "CUDA_VISIBLE_DEVICES="${GPUS}" mpirun -n 1 --allow-run-as-root /opt/tritonserver/bin/tritonserver --model-repository=/model"
     shm_size: '2gb'
     volumes:
@@ -27,7 +27,7 @@ services:
     # For local build
     build:
       context: ../../
-      dockerfile: copilot_proxy/Dockerfile
+      dockerfile: ../../proxy.Dockerfile
     env_file:
       # Automatically created via ./setup.sh
       - test.env

diff --git a/tests/python_backend/docker-compose-without-gpus.yaml b/tests/python_backend/docker-compose-without-gpus.yaml
@@ -3,7 +3,7 @@ services:
   triton:
     build:
       context: ../../
-      dockerfile: Dockerfile
+      dockerfile: ../../triton.Dockerfile
     command: bash -c "CUDA_VISIBLE_DEVICES="${GPUS}" mpirun -n 1 --allow-run-as-root /opt/tritonserver/bin/tritonserver --model-repository=/model"
     shm_size: '2gb'
     volumes:
@@ -20,7 +20,7 @@ services:
     # For local build
     build:
       context: ../../
-      dockerfile: copilot_proxy/Dockerfile
+      dockerfile: ../../proxy.Dockerfile
     env_file:
       # Automatically created via ./setup.sh
       - test.env

diff --git a/tests/python_backend/requirements.txt b/tests/python_backend/requirements.txt
@@ -1,3 +1,3 @@
-pytest
-pexpect
-requests
+pytest==7.2.1
+pexpect==4.8.0
+requests==2.28.2