Ver Fonte

Add Manjaro CUDA include and lib dirs to Makefile (#1212)

Johannes Gäßler há 2 anos atrás
pai
commit
92a6e13a31
1 ficheiros alterados com 2 adições e 2 exclusões
  1. 2 2
      Makefile

+ 2 - 2
Makefile

@@ -105,8 +105,8 @@ ifdef LLAMA_OPENBLAS
 	LDFLAGS += -lopenblas
 	LDFLAGS += -lopenblas
 endif
 endif
 ifdef LLAMA_CUBLAS
 ifdef LLAMA_CUBLAS
-	CFLAGS    += -DGGML_USE_CUBLAS -I/usr/local/cuda/include
-	LDFLAGS   += -lcublas -lculibos -lcudart -lcublasLt -lpthread -ldl -lrt -L/usr/local/cuda/lib64
+	CFLAGS    += -DGGML_USE_CUBLAS -I/usr/local/cuda/include -I$(CUDA_PATH)/targets/x86_64-linux/include
+	LDFLAGS   += -lcublas -lculibos -lcudart -lcublasLt -lpthread -ldl -lrt -L/usr/local/cuda/lib64 -L$(CUDA_PATH)/targets/x86_64-linux/lib
 	OBJS      += ggml-cuda.o
 	OBJS      += ggml-cuda.o
 	NVCC      = nvcc
 	NVCC      = nvcc
 	NVCCFLAGS = --forward-unknown-to-host-compiler -arch=native
 	NVCCFLAGS = --forward-unknown-to-host-compiler -arch=native