From 73aa4e0a33187603af3236243bd250e96e3be878 Mon Sep 17 00:00:00 2001 From: Tim Dettmers Date: Mon, 10 Jul 2023 06:34:04 -0700 Subject: [PATCH] Fixed Makefile and added CUDA 12.2 install. --- Makefile | 4 ++-- cuda_install.sh | 4 ++++ 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/Makefile b/Makefile index 5647fd0..19b5b91 100644 --- a/Makefile +++ b/Makefile @@ -47,8 +47,8 @@ CC_cublasLt110 := -gencode arch=compute_75,code=sm_75 CC_cublasLt110 += -gencode arch=compute_80,code=sm_80 CC_cublasLt111 := -gencode arch=compute_75,code=sm_75 -#CC_cublasLt111 += -gencode arch=compute_80,code=sm_80 -#CC_cublasLt111 += -gencode arch=compute_86,code=sm_86 +CC_cublasLt111 += -gencode arch=compute_80,code=sm_80 +CC_cublasLt111 += -gencode arch=compute_86,code=sm_86 CC_ADA_HOPPER := -gencode arch=compute_89,code=sm_89 CC_ADA_HOPPER += -gencode arch=compute_90,code=sm_90 diff --git a/cuda_install.sh b/cuda_install.sh index 678f7ca..eae2cb5 100644 --- a/cuda_install.sh +++ b/cuda_install.sh @@ -13,6 +13,7 @@ URL117=https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installer URL118=https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run URL120=https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda_12.0.0_525.60.13_linux.run URL121=https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run +URL122=https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run CUDA_VERSION=$1 @@ -65,6 +66,9 @@ if [[ -n "$CUDA_VERSION" ]]; then elif [[ "$CUDA_VERSION" -eq "121" ]]; then URL=$URL121 FOLDER=cuda-12.1 + elif [[ "$CUDA_VERSION" -eq "122" ]]; then + URL=$URL122 + FOLDER=cuda-12.2 else echo "argument error: No cuda version passed as input. Choose among versions 92 to 121" fi