diff --git a/Makefile b/Makefile
index 895f178..5563beb 100644
--- a/Makefile
+++ b/Makefile
@@ -1,10 +1,11 @@
 NVCC = nvcc
+CUDA_ARCH = -arch=sm_86
 
 all: runcuda
 
 .PHONY: runcuda
 runcuda:
-	$(NVCC) -DUSE_CUBLAS=1 -g -o runcuda llama3.cu -lm -lcublas
+	$(NVCC) $(CUDA_ARCH) -DUSE_CUBLAS=1 -g -o runcuda llama3.cu -lm -lcublas
 
 .PHONY: clean
 clean: