diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu index e0d1997de729..49b3557dbb89 100644 --- a/docker/Dockerfile.ci_gpu +++ b/docker/Dockerfile.ci_gpu @@ -137,6 +137,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}