diff --git a/Dockerfile.seldon b/Dockerfile.seldon index 78e5cce4e..e7d412197 100644 --- a/Dockerfile.seldon +++ b/Dockerfile.seldon @@ -2,7 +2,7 @@ FROM axsauze/vulkan-sum:0.1 as vulkan-base # No need to do any work -FROM seldonio/seldon-core-s2i-python3:1.1.1-rc +FROM seldonio/seldon-core-s2i-python3-tf-gpu:1.3.0-dev ENV NVIDIA_DRIVER_CAPABILITIES compute,graphics,utility diff --git a/Makefile b/Makefile index 3709a3fa1..c83f33ac1 100644 --- a/Makefile +++ b/Makefile @@ -53,7 +53,7 @@ docker_seldon_run_gpu: docker_seldon_push: docker push axsauze/vulkan-seldon:0.1 -docker_seldon_build: docker_vulkan_build +docker_seldon_build: docker build . -f Dockerfile.seldon -t axsauze/vulkan-seldon:0.1 docker_vulkan_build: