mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
6bcb545d9c
Make ci cusparselt installation be consistent with nightly binary Remove cu121 related docker build jobs and inductor runs Update test failures relating to cu121 Retry of https://github.com/pytorch/pytorch/pull/145696 Pull Request resolved: https://github.com/pytorch/pytorch/pull/145793 Approved by: https://github.com/eqy, https://github.com/tinglvv
37 lines
1.5 KiB
Bash
37 lines
1.5 KiB
Bash
#!/bin/bash
|
|
|
|
set -ex
|
|
|
|
# cuSPARSELt license: https://docs.nvidia.com/cuda/cusparselt/license.html
|
|
mkdir tmp_cusparselt && cd tmp_cusparselt
|
|
|
|
if [[ ${CUDA_VERSION:0:4} =~ ^12\.[5-8]$ ]]; then
|
|
arch_path='sbsa'
|
|
export TARGETARCH=${TARGETARCH:-$(uname -m)}
|
|
if [ ${TARGETARCH} = 'amd64' ] || [ "${TARGETARCH}" = 'x86_64' ]; then
|
|
arch_path='x86_64'
|
|
fi
|
|
CUSPARSELT_NAME="libcusparse_lt-linux-${arch_path}-0.6.3.2-archive"
|
|
curl --retry 3 -OLs https://developer.download.nvidia.com/compute/cusparselt/redist/libcusparse_lt/linux-${arch_path}/${CUSPARSELT_NAME}.tar.xz
|
|
elif [[ ${CUDA_VERSION:0:4} == "12.4" ]]; then
|
|
arch_path='sbsa'
|
|
export TARGETARCH=${TARGETARCH:-$(uname -m)}
|
|
if [ ${TARGETARCH} = 'amd64' ] || [ "${TARGETARCH}" = 'x86_64' ]; then
|
|
arch_path='x86_64'
|
|
fi
|
|
CUSPARSELT_NAME="libcusparse_lt-linux-${arch_path}-0.6.2.3-archive"
|
|
curl --retry 3 -OLs https://developer.download.nvidia.com/compute/cusparselt/redist/libcusparse_lt/linux-${arch_path}/${CUSPARSELT_NAME}.tar.xz
|
|
elif [[ ${CUDA_VERSION:0:4} == "11.8" ]]; then
|
|
CUSPARSELT_NAME="libcusparse_lt-linux-x86_64-0.4.0.7-archive"
|
|
curl --retry 3 -OLs https://developer.download.nvidia.com/compute/cusparselt/redist/libcusparse_lt/linux-x86_64/${CUSPARSELT_NAME}.tar.xz
|
|
else
|
|
echo "Not sure which libcusparselt version to install for this ${CUDA_VERSION}"
|
|
fi
|
|
|
|
tar xf ${CUSPARSELT_NAME}.tar.xz
|
|
cp -a ${CUSPARSELT_NAME}/include/* /usr/local/cuda/include/
|
|
cp -a ${CUSPARSELT_NAME}/lib/* /usr/local/cuda/lib64/
|
|
cd ..
|
|
rm -rf tmp_cusparselt
|
|
ldconfig
|