Performance-optimized wheels for TensorFlow (SSE, AVX, FMA, XLA, MPI)
CPU wheels for TensorFlow 1.14, built by TinyMind, the cloud machine learning platform. To use the wheels on your own machine, Intel Broadwell or later CPU is required.
We use Compilation flags --config=mkl --copt=-mavx2 --copt=mavx512f --copt=-mfma
in our Broadwell CPU machine. Built for Ubuntu 18.04 LTS .
NOTE: These wheels contain MKL support. If you don't have it, install MKL by following the instructions here.
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.
CPU 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 is required.
We use Compilation flags --config=mkl --copt=-mavx2 --copt=-mfma
in our Broadwell CPU machine. Built for Ubuntu 18.04 LTS .
NOTE: These wheels contain MKL support. If you don't have it, install MKL by following the instructions here.
CPU wheels for TensorFlow 1.12, built by TinyMind, the cloud machine learning platform. To use the wheels on your own machine, Intel Broadwell or later CPU is required.
We use Compilation flags --config=mkl --copt=-mavx2 --copt=-mfma
in our Broadwell CPU machine. Built for Ubuntu 18.04 LTS .
NOTE: These wheels contain MKL support. If you don't have it, install MKL by following the instructions here.
CPU wheels for TensorFlow 1.11, built by TinyMind, the cloud machine learning platform. To use the wheels on your own machine, Intel Broadwell or later CPU is required.
We use Compilation flags --config=mkl --copt=-mavx2 --copt=-mfma
in our Broadwell CPU machine. Built for Ubuntu 18.04 LTS .
NOTE: These wheels contain MKL support. If you don't have it, install MKL by following the instructions here.
GPU wheels for TensorFlow 1.12, 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 -march=native
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.
GPU wheels for TensorFlow 1.12, 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 -march=native
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 is not supported.
GPU wheels for TensorFlow 1.11, 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 -march=native
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.
GPU wheels for TensorFlow 1.11, 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 -march=native
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 is not supported.
GPU wheels for TensorFlow 1.10, 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 -march=native
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.