Skip to content

TensorFlow 1.7 (GPU, CUDA 9, TensorRT, no MKL)

Compare
Choose a tag to compare
@danqing danqing released this 30 Apr 00:35
· 1 commit to master since this release
07cdb25

GPU wheels for TensorFlow 1.7, 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). TensorRT 3.0 is supported.