search

fo40225 / tensorflow-windows-wheel

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

Prebuilt wheel with CUDNN 8 support (either via CUDA 10.2 or 11.0 support) #151

Closed oscarbg closed 3 years ago

fo40225 commented 4 years ago

It is hard to build tf 2.2.0 with cuda 11 & cudnn 8, just wait for tf 2.3.0.

oscarbg commented 4 years ago

ok will wait.. but seems Tensorflow isn’t building 2.3 with CUDA11 & CUDNN 8 by default and seems coming in Tensorflow 2.4.. will you be able to provide an rc1 build to not have to wait some months for final 2.3? thanks..

fo40225 commented 4 years ago

The master branch can be built with cuda 11 & cudnn8, tf2.3 should also work.

I'm trying to cherry-pick some commit to 2.2.0 to build with cuda 11, but there is no good news yet.

oscarbg commented 4 years ago

thanks for trying.. being me wouldn't take the effort to backport changes.. seems tensorflow 2.3 is branching next week so an early build from this branch would be good enough for testing while waiting for a stable release..

fo40225 commented 4 years ago

https://github.com/fo40225/tensorflow-windows-wheel/tree/master/2.2.0/py38/CPU%2BGPU/cuda110cudnn8sse2

tensorflow 2.2.0 + some code fix cuda 11 rc (cuda_11.0.1_451.22) cudnn 8 (cudnn-11.0-v8.0.0.180) python 3.8.0 msvc 2019 16.6 No-AVX

For testing purposes only, no guarantee.

oscarbg commented 3 years ago

thanks! tested &works nice on first try..