From 4e7583eb71bf01a9f341175e139a02f146ef7f5b Mon Sep 17 00:00:00 2001 From: Moritz Maxeiner <moritz@ucworks.org> Date: Tue, 18 Sep 2018 12:44:33 +0200 Subject: [PATCH] Enable cuDNN support for MXNet --- windows/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/windows/Dockerfile b/windows/Dockerfile index c0af797..11fcc8b 100644 --- a/windows/Dockerfile +++ b/windows/Dockerfile @@ -64,7 +64,7 @@ RUN cd $Env:VCPKG_DIR; \ openblas:$vcpkg_triplet \ pybind11:$vcpkg_triplet; \ ./vcpkg install opencv[ffmpeg,ximea,cuda]:$vcpkg_triplet; \ - ./vcpkg install mxnet[cpp,opencv,cuda]:$vcpkg_triplet; \ + ./vcpkg install mxnet[cpp,opencv,cuda,cudnn]:$vcpkg_triplet; \ Copy-Item ./downloads/tools/flang/5.0.0/Library/bin/flang.dll ./installed/$vcpkg_triplet/bin/; \ Copy-Item ./downloads/tools/flang/5.0.0/Library/bin/flangrti.dll ./installed/$vcpkg_triplet/bin/; \ Copy-Item ./downloads/tools/flang/5.0.0/Library/bin/libomp.dll ./installed/$vcpkg_triplet/bin/; \ -- GitLab