Skip to content

Commit 471c95e

Browse files
Yadan-WeiYadan Wei
andauthored
[PATCH] Base CVE Patch (#5018)
* patch base nvjpeg * revert toml --------- Co-authored-by: Yadan Wei <[email protected]>
1 parent 379b51d commit 471c95e

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

scripts/install_cuda.sh

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,19 @@ function prune_cuda {
3333
rm -rf /usr/local/cuda/share/doc
3434
}
3535

36+
#patch CVEs
37+
function install_nvjpeg_for_cuda_below_129 {
38+
mkdir -p /tmp/nvjpeg
39+
cd /tmp/nvjpeg
40+
wget https://developer.download.nvidia.com/compute/cuda/redist/libnvjpeg/linux-x86_64/libnvjpeg-linux-x86_64-12.4.0.76-archive.tar.xz
41+
tar -xvf libnvjpeg-linux-x86_64-12.4.0.76-archive.tar.xz
42+
rm -rf /usr/local/cuda/targets/x86_64-linux/lib/libnvjpeg*
43+
rm -rf /usr/local/cuda/targets/x86_64-linux/include/nvjpeg.h
44+
cp libnvjpeg-linux-x86_64-12.4.0.76-archive/lib/libnvjpeg* /usr/local/cuda/targets/x86_64-linux/lib/
45+
cp libnvjpeg-linux-x86_64-12.4.0.76-archive/include/* /usr/local/cuda/targets/x86_64-linux/include/
46+
rm -rf /tmp/nvjpeg
47+
}
48+
3649

3750
function install_cuda128_stack {
3851
CUDNN_VERSION="9.8.0.87"
@@ -69,6 +82,7 @@ function install_cuda128_stack {
6982
cp -a build/include/* /usr/local/cuda/include/
7083
cp -a build/lib/* /usr/local/cuda/lib64/
7184

85+
install_nvjpeg_for_cuda_below_129
7286
prune_cuda
7387
ldconfig
7488
}

0 commit comments

Comments
 (0)