search

fo40225 / tensorflow-windows-wheel

Tensorflow prebuilt binary for Windows
3.66k stars 1.53k forks source link

Cuda 11 cuDNN 8 #156

Closed Expert73 closed 4 years ago

Expert73 commented 4 years ago

how did you solve the problem with the thust module during the build?

when i try to build i have some problem:

error The version of CUB in your include path is not compatible with this release of Thrust. CUB is now included in the CUDA Toolkit, so you no longer need to use your own checkout of CUB. Define THRUST_IGNORE_CUB_VERSION_CHECK to ignore this.

fo40225 commented 4 years ago 
Please specify optimization flags to use during compilation when bazel option "--config=opt" is specified:
/DTHRUST_IGNORE_CUB_VERSION_CHECK
Expert73 commented 4 years ago

I don't understand what the command line will look like. something like this? Please specify optimization flags to use during compilation when bazel option "--config=opt" is specified: /DTHRUST_IGNORE_CUB_VERSION_CHECK /arch:AVX2 or /arch:AVX2 not whrite?

Expert73 commented 4 years ago

I think TF 2.3 not work with cudnn 8 (final). if you can make the build share the secret of how you did it.

fo40225 commented 4 years ago

I can build tf 2.3.0 with cuda11+cudnn8 without modification.

Expert73 commented 4 years ago

thanks! I try it again. Maybe my problem in mkl. I build versions 2.1 and 2.2 with double optimizations. The performance was higher