search

yaroslavvb / tensorflow-community-wheels

Place to upload links to TensorFlow wheels
453 stars 35 forks source link

Tensorflow 1.2.0 Win64, GPU and AVX enabled #24

Open aluo-x opened 7 years ago

aluo-x commented 7 years ago

More details are in the readme. https://github.com/aluo-x/tensorflow_windows

bconstanzo commented 7 years ago

This would be great for Python 3.6! I'll be checking around and test it the moment you get it on 3.6 :)

aluo-x commented 7 years ago

I have discovered that batchnorm is broken is the current build. I would advise against using my build until further investigation. A discussion can be found in my repo.

bconstanzo commented 7 years ago

Thanks for the warning, though for the moment I'm using Crhistoph Gohlke's build (Windows, Py3.6, no SSE, no nothing, but works).

bconstanzo commented 7 years ago

I see in the repo that you've been working on this. Here's hoping that you figure the broken things on the 3.6 builds -- I'll test as soon as I you upload, and thanks for the hard work!

aluo-x commented 7 years ago

So I filed an issue here #11865, awaiting a response.

aluo-x commented 7 years ago

I've added a CPU only version w/ AVX2, built using VS 2017 15.4 and Python 3.6.3. So far GPU versions with CUDA 9 and cuDNN 7 are still failing on Windows, but will likely be added with TF 1.5.

Please note AVX2 will only work on Intel Haswell and later, and AMD Excavator & Zen.

aluo-x commented 6 years ago

just uploaded a whl with CUDA 8 + cudnn 7 + AVX (not 2) support.

bconstanzo commented 6 years ago

You sir, are a gentleman. I'll be checking (and most probably using) those AVX builds!