Skip to content

TensorFlow 1.13 (GPU, CUDA 10.0, cuDNN 7.4, TensorRT, no MKL)

Compare
Choose a tag to compare
@BobDu BobDu released this 01 Mar 06:21
236b106

GPU wheels for TensorFlow 1.13, 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 10.0 and cuDNN 7.4 are required.

We use Compilation flags --copt=-mavx in our Broadwell CPU machine. Built for Ubuntu 18.04 LTS .

This version is optimized for compute capabilities 3.7 (K80, AWS P2/GCP), 6.0 (P100, GCP), 6.1(GTX1050/1060/1080/1080Ti) and 7.0 (V100, AWS P3).

TensorRT 5.0 is supported.