Hi,
We found a workaround to unblock this issue.
Please use basic cudnnConvolutionForward algorithm instead.
1. Download source
wget http://dlib.net/files/dlib-19.16.tar.bz2
tar jxvf dlib-19.16.tar.bz2
2. Apply this changes:
diff --git a/dlib/cuda/cudnn_dlibapi.cpp b/dlib/cuda/cudnn_dlibapi.cpp
index a32fcf6..6952584 100644
--- a/dlib/cuda/cudnn_dlibapi.cpp
+++ b/dlib/cuda/cudnn_dlibapi.cpp
@@ -851,7 +851,7 @@ namespace dlib
dnn_prefer_fastest_algorithms()?CUDNN_CONVOLUTION_FWD_PREFER_FASTEST:CUDNN_CONVOLUTION_FWD_NO_WORKSPACE,
std::numeric_limits<size_t>::max(),
&forward_best_algo));
- forward_algo = forward_best_algo;
+ //forward_algo = forward_best_algo;
CHECK_CUDNN(cudnnGetConvolutionForwardWorkspaceSize(
context(),
descriptor(data),
3. Build and install
mkdir build
cd build
cmake ..
cmake --build .
sudo python setup.py install
Our internal team keep checking the cuDNN issue and will let you know if any progress.
Thanks.