Skip to content

TensorFlow 1.3 (GPU, MPI)

Compare
Choose a tag to compare
@danqing danqing released this 21 Aug 06:02
· 42 commits to master since this release

GPU wheels for TensorFlow 1.3, 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. This version contains MPI support. If you are having issues with MPI, make sure you have mpich installed via:

sudo apt-get install mpich

If you don't need MPI support, you may want to use the non-MPI version.