File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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
4747pip3 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+
4961rm -fr * .log ./models
5062
5163mkdir -p models/dlpack_test/1/
You can’t perform that action at this time.
0 commit comments