@@ -64,12 +64,19 @@ for python_version in "3.7" "3.8"; do
6464
6565 echo " Adding OneDNN to the dockerfile"
6666 echo " # Adding MKL-DNN (now OneDNN) to the image" >> " ${output_file} "
67- apt_install_temp_packages ${output_file} " ${BUILD_PACKAGES} "
68- echo " && git clone https://github.com/01org/mkl-dnn.git -b ${ONE_DNN_VERSION} --depth 1 \\ " >> " ${output_file} "
69- echo " && cd mkl-dnn && mkdir -p build && cd build \\ " >> " ${output_file} "
70- echo " && cmake .. \\ " >> " ${output_file} "
71- echo " && make -j \\ " >> " ${output_file} "
72- echo " && make install \\ " >> " ${output_file} "
67+ apt_install_temp_packages ${output_file} " curl"
68+ # https://github.com/oneapi-src/oneDNN/releases/download/v1.3/dnnl_lnx_1.3.0_cpu_gomp.tgz
69+ echo " && curl -L https://github.com/oneapi-src/oneDNN/releases/download/v1.3/dnnl_lnx_1.3.0_cpu_gomp.tgz -o dnnl.tgz \\ " >> " ${output_file} "
70+ echo " && tar zxvf dnnl.tgz \\ " >> " ${output_file} "
71+ echo " && mv dnnl_lnx_1.3.0_cpu_gomp/include/* /usr/local/include \\ " >> " ${output_file} "
72+ echo " && mv dnnl_lnx_1.3.0_cpu_gomp/lib/* /usr/local/lib \\ " >> " ${output_file} "
73+ echo " && rm dnnl.tgz && rm -r dnnl_lnx_1.3.0_cpu_gomp \\ " >> " ${output_file} "
74+
75+ # echo "&& git clone https://github.com/01org/mkl-dnn.git -b ${ONE_DNN_VERSION} --depth 1 \\" >>"${output_file}"
76+ # echo "&& cd mkl-dnn && mkdir -p build && cd build \\">>"${output_file}"
77+ # echo "&& cmake .. \\">>"${output_file}"
78+ # echo "&& make -j \\">>"${output_file}"
79+ # echo "&& make install \\">>"${output_file}"
7380 # echo "&& cd mkl-dnn/scripts && ./prepare_mkl.sh && cd .. \\" >>"${output_file}"
7481 # echo "&& mkdir -p build && cd build && cmake .. && make \\" >>"${output_file}"
7582 # echo "&& make install \\" >>"${output_file}"
0 commit comments