[ci][docker gpu] Install dnnl in docker GPU. (#11744)

BYOC related tutorial may use dnnl and such tutorial run at docker gpu which need to install dnnl to prepare the environment.
apache · Jun 17, 2022 · 1b8f3b5 · 1b8f3b5
1 parent 7433b2f
commit 1b8f3b5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu
@@ -139,6 +139,10 @@ COPY install/ubuntu_install_sccache.sh /install/ubuntu_install_sccache.sh
 RUN bash /install/ubuntu_install_sccache.sh
 ENV PATH /opt/sccache:$PATH
 
+# dnnl
+COPY install/ubuntu_install_dnnl.sh /install/ubuntu_install_dnnl.sh
+RUN bash /install/ubuntu_install_dnnl.sh
+
 # Environment variables
 ENV PATH=/usr/local/nvidia/bin:${PATH}
 ENV PATH=/usr/local/cuda/bin:${PATH}