Skip to content

Commit 94fc4c6

Browse files
committed
temp: FIx for library absence (#8355)
1 parent c4abbec commit 94fc4c6

1 file changed

Lines changed: 12 additions & 0 deletions

File tree

qa/L0_dlpack_multi_gpu/test.sh

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,18 @@ pip3 install torch==2.3.1+cu118 -f https://download.pytorch.org/whl/torch_stable
4646
# Install CuPy for testing non_blocking compute streams
4747
pip3 install cupy-cuda12x
4848

49+
if [ ${CUDA_VERSION%%.*} -gt 12 ]; then
50+
curl -L https://developer.download.nvidia.com/compute/cuda/redist/cuda_nvrtc/linux-x86_64/cuda_nvrtc-linux-x86_64-12.9.86-archive.tar.xz \
51+
-o /tmp/cuda_nvrtc-linux-x86_64-12.9.86-archive.tar.xz ;
52+
curl -L https://developer.download.nvidia.com/compute/cuda/redist/libcublas/linux-x86_64/libcublas-linux-x86_64-12.9.1.4-archive.tar.xz \
53+
-o /tmp/libcublas-linux-x86_64-12.9.1.4-archive.tar.xz ;
54+
cd /tmp ;
55+
tar -xvf /tmp/cuda_nvrtc-linux-x86_64-12.9.86-archive.tar.xz --strip-components=1 ;
56+
tar -xvf /tmp/libcublas-linux-x86_64-12.9.1.4-archive.tar.xz --strip-components=1 ;
57+
export LD_LIBRARY_PATH=/tmp/lib:$LD_LIBRARY_PATH ;
58+
cd -
59+
fi
60+
4961
rm -fr *.log ./models
5062

5163
mkdir -p models/dlpack_test/1/

0 commit comments

Comments
 (0)