From 44348d68cf8357d420b4dcf66d0c23a4891dc0e2 Mon Sep 17 00:00:00 2001 From: Joywalker Date: Wed, 21 Feb 2024 22:35:24 +0200 Subject: [PATCH] fix: Added check nvidia in Makefile --- .../Makefile | 16 ++++++++++++++++ .../{blueprint.json => od_blueprint.json} | 0 2 files changed, 16 insertions(+) rename cvision/deep_learning_model_optimization_pipeline/{blueprint.json => od_blueprint.json} (100%) diff --git a/cvision/deep_learning_model_optimization_pipeline/Makefile b/cvision/deep_learning_model_optimization_pipeline/Makefile index f6e68ea..bb4c648 100644 --- a/cvision/deep_learning_model_optimization_pipeline/Makefile +++ b/cvision/deep_learning_model_optimization_pipeline/Makefile @@ -11,10 +11,26 @@ SHELL := /bin/bash help: @echo -e "$(BLUE)Usage: make [target]$(NC)" @echo -e "$(BLUE)Targets:$(NC)" + @echo -e "$(BLUE) check_nvidia$(NC) - Check for NVIDIA Container Toolkit and nvcc" @echo -e "$(BLUE) install_nvidia_container_runtime$(NC) - Install NVIDIA Container Runtime" @echo -e "$(BLUE) install_nvidia_cuda_cc$(NC) - Install NVIDIA CUDA" + +check_nvidia: + @echo -e "$(BLUE)Checking for NVIDIA Container Toolkit...$(NC)" + @if dpkg -l | grep -qw nvidia-container-toolkit; then \ + echo -e "$(GREEN)NVIDIA Container Toolkit is already installed.$(NC)"; \ + else \ + echo -e "$(RED)NVIDIA Container Toolkit is not installed.$(NC)"; \ + fi + @echo -e "$(BLUE)Checking for nvcc...$(NC)" + @if hash nvcc 2>/dev/null; then \ + echo -e "$(GREEN)nvcc is already installed. Version: $$(nvcc --version | grep release | sed 's/.*release //;s/,.*//')$(NC)"; \ + else \ + echo -e "$(RED)nvcc is not installed.$(NC)"; \ + fi + install_nvidia_container_runtime: @echo -e "$(GREEN)Installing NVIDIA Container Runtime...$(NC)" @curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \ diff --git a/cvision/deep_learning_model_optimization_pipeline/blueprint.json b/cvision/deep_learning_model_optimization_pipeline/od_blueprint.json similarity index 100% rename from cvision/deep_learning_model_optimization_pipeline/blueprint.json rename to cvision/deep_learning_model_optimization_pipeline/od_blueprint.json