File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff line change @@ -151,7 +151,11 @@ if [[ $CUDA_VERSION == "12.1" || $CUDA_VERSION == "12.4" ]]; then
151
151
DEPS_LIST+=(
152
152
" /usr/local/cuda/lib64/libcudnn_adv.so.9"
153
153
" /usr/local/cuda/lib64/libcudnn_cnn.so.9"
154
+ " /usr/local/cuda/lib64/libcudnn_graph.so.9"
154
155
" /usr/local/cuda/lib64/libcudnn_ops.so.9"
156
+ " /usr/local/cuda/lib64/libcudnn_engines_runtime_compiled.so.9"
157
+ " /usr/local/cuda/lib64/libcudnn_engines_precompiled.so.9"
158
+ " /usr/local/cuda/lib64/libcudnn_heuristic.so.9"
155
159
" /usr/local/cuda/lib64/libcudnn.so.9"
156
160
" /usr/local/cuda/lib64/libcublas.so.12"
157
161
" /usr/local/cuda/lib64/libcublasLt.so.12"
@@ -163,7 +167,11 @@ if [[ $CUDA_VERSION == "12.1" || $CUDA_VERSION == "12.4" ]]; then
163
167
DEPS_SONAME+=(
164
168
" libcudnn_adv.so.9"
165
169
" libcudnn_cnn.so.9"
170
+ " libcudnn_graph.so.9"
166
171
" libcudnn_ops.so.9"
172
+ " libcudnn_engines_runtime_compiled.so.9"
173
+ " libcudnn_engines_precompiled.so.9"
174
+ " libcudnn_heuristic.so.9"
167
175
" libcudnn.so.9"
168
176
" libcublas.so.12"
169
177
" libcublasLt.so.12"
You can’t perform that action at this time.
0 commit comments