Skip to content

TensorFlow 1.4.1 (GPU)

Compare
Choose a tag to compare
@danqing danqing released this 09 Dec 05:50
· 14 commits to master since this release
1bf7e7b

GPU wheels for TensorFlow 1.4.1, 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 8 and CuDNN 6 are required.

NOTE: Starting from TF 1.5, we will move to provide CUDA 9 wheels only. You should be able to upgrade your system to CUDA 9 and enjoy the performance boost. If you need CUDA 8 wheels for future versions, file an issue. If there is enough demand, we may consider keeping CUDA 8 builds around.

For TF 1.4.1, you can find CUDA 9 wheels here.

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