ADded gpu rules to makefile

2020-08-05 11:37:23 +01:00 · 2020-08-05 11:37:23 +01:00 · 2dd21a75cd
commit 2dd21a75cd
parent e874e878ec
1 changed files with 12 additions and 1 deletions
--- a/13
+++ b/13
@ -37,7 +37,18 @@ build_shaders:

 docker_seldon_run:
 	docker run \
-		--rm -it -p 5000:5000 axsauze/vulkan-seldon:0.1 seldon-core-microservice Model REST
+		--rm -it \
+		-p 5000:5000 \
+		axsauze/vulkan-seldon:0.1 \
+		seldon-core-microservice Model REST
+
+docker_seldon_run_gpu:
+	docker run \
+		--rm -it \
+		--gpus all \
+		-p 5000:5000 \
+		axsauze/vulkan-seldon:0.1 \
+		seldon-core-microservice Model REST

 docker_seldon_push:
 	docker push axsauze/vulkan-seldon:0.1