File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -116,7 +116,7 @@ cd "$scriptDir/.."
116
116
# Dependency files
117
117
118
118
# MKL
119
- declare -a mklFiles=(" libmklml_intel.so" " libiomp5.so" " libmkldnn.so.0.12.0 " )
119
+ declare -a mklFiles=(" libmklml_intel.so" " libiomp5.so" " libmkldnn.so.0" )
120
120
121
121
# Open CV
122
122
declare -a opencvFiles=(" libopencv_core.so.3.1" " libopencv_imgproc.so.3.1" " libopencv_imgproc.so.3.1" " libopencv_imgcodecs.so.3.1" )
@@ -276,7 +276,7 @@ if [[ $targetConfiguration != "cpu" ]]; then
276
276
277
277
# Copy cuDNN
278
278
echo " Copying cuDNN..." >&3
279
- CopyFilesFromList $cudnnPath cudnnFiles[@] $baseDependenciesPath /libcudnn.so.6
279
+ CopyFilesFromList $cudnnPath cudnnFiles[@] $baseDependenciesPath /libcudnn.so.7
280
280
281
281
# Copy NCCL
282
282
echo " Copying NCCL..." >&3
You can’t perform that action at this time.
0 commit comments