Skip to content

TensorFlow 1.4 (GPU, CUDA 9, Compute 3.7)

Compare
Choose a tag to compare
@danqing danqing released this 04 Nov 21:40
· 26 commits to master since this release
7be7f68

GPU wheels for TensorFlow 1.4, built by TinyMind, the cloud machine learning platform.

To use the wheels on your own machine, Intel Broadwell or later CPU, and Nvidia computing capability 3.7 or later GPU with CUDA 9 and CuDNN 7 are required.

If you are using Nvidia P100 or V100 GPU, you may want to use these optimized wheels instead.

NOTE: These wheels contain MKL support. If you don't have it, install MKL by following the instructions here.