Skip to content

Commit 1b8f3b5

Browse files
authored
[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.
1 parent 7433b2f commit 1b8f3b5

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

docker/Dockerfile.ci_gpu

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -139,6 +139,10 @@ COPY install/ubuntu_install_sccache.sh /install/ubuntu_install_sccache.sh
139139
RUN bash /install/ubuntu_install_sccache.sh
140140
ENV PATH /opt/sccache:$PATH
141141

142+
# dnnl
143+
COPY install/ubuntu_install_dnnl.sh /install/ubuntu_install_dnnl.sh
144+
RUN bash /install/ubuntu_install_dnnl.sh
145+
142146
# Environment variables
143147
ENV PATH=/usr/local/nvidia/bin:${PATH}
144148
ENV PATH=/usr/local/cuda/bin:${PATH}

0 commit comments

Comments
 (0)