summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorkwin1412 <42286931+kwin1412@users.noreply.github.com>2024-02-26 00:46:49 +0800
committerGitHub <noreply@github.com>2024-02-25 18:46:49 +0200
commitf1a98c52546d009f742bdec2154c2a314ea950a6 (patch)
treec0cb643b1118e5d77603b529f2d1122beef83863
parent7d548a1827f6fc6aece6db74c9d112da42c40d68 (diff)
make : fix nvcc version is empty (#5713)
fix nvcc version is empty
-rw-r--r--Makefile2
1 files changed, 1 insertions, 1 deletions
diff --git a/Makefile b/Makefile
index f03faf6e..068f6ed0 100644
--- a/Makefile
+++ b/Makefile
@@ -597,7 +597,7 @@ $(info I CC: $(shell $(CC) --version | head -n 1))
$(info I CXX: $(shell $(CXX) --version | head -n 1))
ifdef LLAMA_CUBLAS
$(info I NVCC: $(shell $(NVCC) --version | tail -n 1))
-CUDA_VERSION := $(shell nvcc --version | grep -oP 'release (\K[0-9]+\.[0-9])')
+CUDA_VERSION := $(shell $(NVCC) --version | grep -oP 'release (\K[0-9]+\.[0-9])')
ifeq ($(shell awk -v "v=$(CUDA_VERSION)" 'BEGIN { print (v < 11.7) }'),1)
ifndef CUDA_DOCKER_ARCH
ifndef CUDA_POWER_ARCH