diff --git a/Makefile b/Makefile index ecb6c9011..892f17e70 100644 --- a/Makefile +++ b/Makefile @@ -141,7 +141,7 @@ endif # it is recommended to use the CMAKE file to build for cublas if you can - will likely work better ifdef LLAMA_CUBLAS CUBLAS_FLAGS = -DGGML_USE_CUBLAS -I/usr/local/cuda/include -I/opt/cuda/include -I$(CUDA_PATH)/targets/x86_64-linux/include - CUBLASLD_FLAGS = -lcuda -lcublas -lculibos -lcudart -lcublasLt -lpthread -ldl -lrt -L/usr/local/cuda/lib64 -L/opt/cuda/lib64 -L$(CUDA_PATH)/targets/x86_64-linux/lib -L/usr/local/cuda/targets/aarch64-linux/lib -L/usr/lib/wsl/lib + CUBLASLD_FLAGS = -lcuda -lcublas -lcudart -lcublasLt -lpthread -ldl -lrt -L/usr/local/cuda/lib64 -L/opt/cuda/lib64 -L$(CUDA_PATH)/targets/x86_64-linux/lib -L/usr/local/cuda/targets/aarch64-linux/lib -L/usr/lib/wsl/lib CUBLAS_OBJS = ggml-cuda.o ggml_v3-cuda.o ggml_v2-cuda.o ggml_v2-cuda-legacy.o NVCC = nvcc NVCCFLAGS = --forward-unknown-to-host-compiler -use_fast_math diff --git a/environment.yaml b/environment.yaml index cc9e1fcc2..8ffc0ba04 100644 --- a/environment.yaml +++ b/environment.yaml @@ -5,8 +5,8 @@ channels: - defaults dependencies: - python=3.8 - - cuda-nvcc=11.5 - - cuda-libraries-dev=11.5 + - cuda-nvcc + - cuda-libraries-dev - cxx-compiler - gxx=10 - pip