File tree Expand file tree Collapse file tree 1 file changed +14
-0
lines changed Expand file tree Collapse file tree 1 file changed +14
-0
lines changed Original file line number Diff line number Diff line change @@ -33,6 +33,19 @@ function prune_cuda {
33
33
rm -rf /usr/local/cuda/share/doc
34
34
}
35
35
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
+
36
49
37
50
function install_cuda128_stack {
38
51
CUDNN_VERSION=" 9.8.0.87"
@@ -69,6 +82,7 @@ function install_cuda128_stack {
69
82
cp -a build/include/* /usr/local/cuda/include/
70
83
cp -a build/lib/* /usr/local/cuda/lib64/
71
84
85
+ install_nvjpeg_for_cuda_below_129
72
86
prune_cuda
73
87
ldconfig
74
88
}
You can’t perform that action at this time.
0 commit comments