Mind Wheels Versions Save

Performance-optimized wheels for TensorFlow (SSE, AVX, FMA, XLA, MPI)

tf1.5-gpu-generic

6 years ago

CUDA 9.1 wheels for TensorFlow 1.5, without optimization. We provide them here as TensorFlow hasn't officially provided CUDA 9.1 wheels, and our optimized builds may not work for AMD or old Intel CPUs.

tf1.5-gpu-cuda91

6 years ago

GPU wheels for TensorFlow 1.5, 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.1 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).

NOTE: These wheels contain MKL support. If you don't have it, install MKL by following the instructions here. You may want to take a look at the performance guide for MKL.

tf1.5-cpu-mkl

6 years ago

CPU wheels for TensorFlow 1.5 with MKL. You may want to take a look at the performance guide for MKL.

NOTE: These wheels contain MKL support. If you don't have it, install MKL by following the instructions here.

You can find the non-MKL CPU version for TensorFlow 1.5 here.

tf1.5-gpu-cuda91-nomkl

6 years ago

GPU wheels for TensorFlow 1.5, 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.1 and CuDNN 7.1.2 are required.

This version is optimized for compute capabilities 3.7 (K80, AWS P2/GCP).

tf1.5-cpu

6 years ago

CPU wheels for TensorFlow 1.5, built by TinyMind, the cloud machine learning platform.

To use the wheels on your own machine, Intel Broadwell or later CPU is required.

tf1.4.1-gpu-cuda91-generic

6 years ago

CUDA 9.1 wheels for TensorFlow 1.4.1, without optimization. We provide them here as TensorFlow hasn't officially provided CUDA 9.1 wheels, and our optimized builds may not work for AMD or old Intel CPUs.

tf-adhoc

6 years ago

Adhoc builds requested by users.

If you have a request, file an issue and we will see if we can get a build for you.

  • tensorflow-cuda8-1.4.1-cp36-cp36m-linux_x86_64.whl: Python 3.6, CUDA 8/cuDNN 6 without MKL, Ubuntu 16.04
  • tensorflow-cuda91-1.4.1-cp36-cp36m-linux_x86_64.whl: Python 3.6, CUDA 9.1/cuDNN 7 with Compute 3.5, MKL, Ubuntu 16.04

tf1.4.1-gpu-cuda91

6 years ago

GPU wheels for TensorFlow 1.4.1, 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.1 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).

NOTE: These wheels contain MKL support. If you don't have it, install MKL by following the instructions here. You may want to take a look at the performance guide for MKL.

tf1.4.1-cpu

6 years ago

CPU wheels for TensorFlow 1.4.1, built by TinyMind, the cloud machine learning platform.

To use the wheels on your own machine, Intel Broadwell or later CPU is required.

tf1.4.1-gpu-cuda9

6 years ago

GPU wheels for TensorFlow 1.4.1, 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).

NOTE: These wheels contain MKL support. If you don't have it, install MKL by following the instructions here.