Skip to content

Commit

Permalink
Revert "Add cuSparseLt-0.5.0 to manywheel images"
Browse files Browse the repository at this point in the history
This reverts commit 00841b6 as
cuSparseLT is not compatible with CentOS 7
  • Loading branch information
malfet committed Oct 11, 2023
1 parent 321ab64 commit 4e68b6d
Showing 1 changed file with 2 additions and 22 deletions.
24 changes: 2 additions & 22 deletions common/install_cuda.sh
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
set -ex

function install_118 {
echo "Installing CUDA 11.8 and cuDNN 8.7 and NCCL 2.15 and cuSparseLt-0.5.0"
echo "Installing CUDA 11.8 and cuDNN 8.7 and NCCL 2.15"
rm -rf /usr/local/cuda-11.8 /usr/local/cuda
# install CUDA 11.8.0 in the same container
wget -q https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
Expand Down Expand Up @@ -31,20 +31,10 @@ function install_118 {
cd ..
rm -rf tmp_nccl
ldconfig

# cuSparseLt license: https://docs.nvidia.com/cuda/cusparselt/license.html
mkdir tmp_cusparselt && pushd tmp_cusparselt
wget -q https://developer.download.nvidia.com/compute/cusparselt/redist/libcusparse_lt/linux-x86_64/libcusparse_lt-linux-x86_64-0.5.0.1-archive.tar.xz
tar xf libcusparse_lt-linux-x86_64-0.5.0.1-archive.tar.xz
cp -a libcusparse_lt-linux-x86_64-0.5.0.1-archive/include/* /usr/local/cuda/include/
cp -a libcusparse_lt-linux-x86_64-0.5.0.1-archive/lib/* /usr/local/cuda/lib64/
popd
rm -rf tmp_custparselt
ldconfig
}

function install_121 {
echo "Installing CUDA 12.1 and cuDNN 8.9 and NCCL 2.18.1 and cuSparseLt-0.5.0"
echo "Installing CUDA 12.1 and cuDNN 8.9 and NCCL 2.18.1"
rm -rf /usr/local/cuda-12.1 /usr/local/cuda
# install CUDA 12.1.0 in the same container
wget -q https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda_12.1.1_530.30.02_linux.run
Expand Down Expand Up @@ -72,16 +62,6 @@ function install_121 {
cd ..
rm -rf tmp_nccl
ldconfig

# cuSparseLt license: https://docs.nvidia.com/cuda/cusparselt/license.html
mkdir tmp_cusparselt && pushd tmp_cusparselt
wget -q https://developer.download.nvidia.com/compute/cusparselt/redist/libcusparse_lt/linux-x86_64/libcusparse_lt-linux-x86_64-0.5.0.1-archive.tar.xz
tar xf libcusparse_lt-linux-x86_64-0.5.0.1-archive.tar.xz
cp -a libcusparse_lt-linux-x86_64-0.5.0.1-archive/include/* /usr/local/cuda/include/
cp -a libcusparse_lt-linux-x86_64-0.5.0.1-archive/lib/* /usr/local/cuda/lib64/
popd
rm -rf tmp_custparselt
ldconfig
}

function prune_118 {
Expand Down

0 comments on commit 4e68b6d

Please sign in to comment.