Skip to content

GitLab

Commit 73aa4e0a authored by Tim Dettmers's avatar Tim Dettmers
Browse files

Fixed Makefile and added CUDA 12.2 install.

parent 5f492d43
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 2 deletions
Makefile
View file @ 73aa4e0a
...@@ -47,8 +47,8 @@ CC_cublasLt110 := -gencode arch=compute_75,code=sm_75 ...@@ -47,8 +47,8 @@ CC_cublasLt110 := -gencode arch=compute_75,code=sm_75
CC_cublasLt110 += -gencode arch=compute_80,code=sm_80 CC_cublasLt110 += -gencode arch=compute_80,code=sm_80
CC_cublasLt111 := -gencode arch=compute_75,code=sm_75 CC_cublasLt111 := -gencode arch=compute_75,code=sm_75
#CC_cublasLt111 += -gencode arch=compute_80,code=sm_80 CC_cublasLt111 += -gencode arch=compute_80,code=sm_80
#CC_cublasLt111 += -gencode arch=compute_86,code=sm_86 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_89,code=sm_89
CC_ADA_HOPPER += -gencode arch=compute_90,code=sm_90 CC_ADA_HOPPER += -gencode arch=compute_90,code=sm_90
......
cuda_install.sh
View file @ 73aa4e0a
...@@ -13,6 +13,7 @@ URL117=https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installer ...@@ -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 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 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 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 CUDA_VERSION=$1
...@@ -65,6 +66,9 @@ if [[ -n "$CUDA_VERSION" ]]; then ...@@ -65,6 +66,9 @@ if [[ -n "$CUDA_VERSION" ]]; then
elif [[ "$CUDA_VERSION" -eq "121" ]]; then elif [[ "$CUDA_VERSION" -eq "121" ]]; then
URL=$URL121 URL=$URL121
FOLDER=cuda-12.1 FOLDER=cuda-12.1
elif [[ "$CUDA_VERSION" -eq "122" ]]; then
URL=$URL122
FOLDER=cuda-12.2
else else
echo "argument error: No cuda version passed as input. Choose among versions 92 to 121" echo "argument error: No cuda version passed as input. Choose among versions 92 to 121"
fi fi
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment