TensorFlow 1.4 (GPU, CUDA 9, Compute 3.7/6.0/7.0, no MKL)
amberfly
released this
22 Nov 22:06
·
16 commits
to master
since this release
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.
This version is optimized for compute capabilities 3.7 (K80, AWS P2/GCP), 6.0 (P100, GCP) and 7.0 (V100, AWS P3).
These wheels don't contain MKL support. It's recommended that you install MKL to your system to speed up TensorFlow unless you are unable to do so. Wheels with MKL support can be found at:
- CUDA 9 + Compute 3.7 if you have Nvidia K80 or equivalent GPU, or
- CUDA 9 + Compute 3.7, 6.0, 7.0 if you have Nvidia P100, V100 or equivalent GPU.