TensorRT was linked against cudnn 7.6.3 but loaded cudnn 7.4

when I building up the TensorRT in my computer, after configure in windows, I use the VS2017 to rebuild the mnist sample, then run the generataed exe, it shows can't find cudnn64_7.dll. So search this file in my computer, and copy one to this bin path, then

I meet this warning:

TensorRT was linked against cudnn 7.6.3 but loaded cudnn 7.4

Environment:

  • Windows 10 Home version
  • VS2017
  • GTX1050 Max, cuda 10.0, cudnn7.6.3

Reason, there are two same name cudnn64_7.dll in my computer, one is for torch, and another is for C++, unfortunately, I copied the torch version, so use the C++ version instead of it, then the warning is gone.

Another Tips: there is issue for run the mnist samples due to the *.pgm files can't find. this becasue from 6.0, the *pgm files removed, it seems that we need to download the MNIST data manually, and the use gunzip to unzip the file to get train-labels-idx1-ubyte file. Finally, run generate_pgms.py file to get the picture file(*.pgm files).

There is a file named download_pgms.py in the data folder. So just run it to got the test smaples. If all goes well, you wil got follows result:

TensorRT was linked against cudnn 7.6.3 but loaded cudnn 7.4_第1张图片

你可能感兴趣的:(C++,深度学习,深度学习)