1#!/bin/bash 2 3set -ex 4 5# cudss license: https://docs.nvidia.com/cuda/cudss/license.html 6mkdir tmp_cudss && cd tmp_cudss 7 8if [[ ${CUDA_VERSION:0:4} =~ ^12\.[1-4]$ ]]; then 9 arch_path='sbsa' 10 export TARGETARCH=${TARGETARCH:-$(uname -m)} 11 if [ ${TARGETARCH} = 'amd64' ] || [ "${TARGETARCH}" = 'x86_64' ]; then 12 arch_path='x86_64' 13 fi 14 CUDSS_NAME="libcudss-linux-${arch_path}-0.3.0.9_cuda12-archive" 15 curl --retry 3 -OLs https://developer.download.nvidia.com/compute/cudss/redist/libcudss/linux-${arch_path}/${CUDSS_NAME}.tar.xz 16 17 # only for cuda 12 18 tar xf ${CUDSS_NAME}.tar.xz 19 cp -a ${CUDSS_NAME}/include/* /usr/local/cuda/include/ 20 cp -a ${CUDSS_NAME}/lib/* /usr/local/cuda/lib64/ 21fi 22 23cd .. 24rm -rf tmp_cudss 25ldconfig 26