If you find any of this useful, consider being my sponsor. I really appreciate it, or get me some coffee, coffee is great! Check out the sponsor tiers or the external links on the sidebar for one time donations.
Tensorflow wheels I've been building and using over time. Unless specified, these are built for skylake CPUs or equivalent instruction sets. This means they support SSE4.2 and AVX2 and all the other instructions supported by skylake microarchitecture. Most builds have a minimum of CUDA compute capability of 6.1.
Although builds are made in Arch Linux, the latest wheel is fully compatible with Ubuntu 20.04 or 20.10 (depending on your glibc version), other wheels are compatible with 18.04, which comes with older nvidia drivers. You can see the compatibility between drivers and CUDA in the CUDA Driver section
| TF | HW | Python | CUDA | cuDNN | Support | OS | Download |
|---|---|---|---|---|---|---|---|
| 2.5.0 | GPU | 3.9 | 11 | 8 | XLA,SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 2.5.0rc3 | GPU | 3.9 | 11 | 8 | XLA,SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 2.4.1 | CPU | 3.9 | - | - | XLA,SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 2.4.1 | GPU | 3.8 | 11.1 | 8 | XLA,SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 2.4 | GPU | 3.8 | 11.1 | 8 | XLA,SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 2.4rc4 | CPU | 3.8 | 11.1 | 8 | TensorRT, XLA,SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 2.3 | GPU | 3.8 | 11 | 8 | XLA,SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 2.2 | GPU | 3.8 | 10.2 | 7.6 | XLA,SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 2.2rc2 | GPU | 3.7 | 10.2 | 7.6 | XLA,SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 2.1 | GPU | 3.7 | 10.2 | 7.6 | XLA,SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 2.1 | GPU | 3.7 | 10.1 | 7.6 | XLA,SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 2.0 | GPU | 3.7 | 10.1 | 7.6 | XLA,SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 2.0rc2 | GPU | 3.7 | 10.1 | 7.6 | XLA,SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 2.0rc2 | GPU | 3.7 | 10.1 | 7.6 | SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 1.14 | GPU | 3.7 | 10.0 | 7.4 | XLA,SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 1.13.1 | GPU | 3.7 | 10.0 | 7.4 | XLA,SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 1.13.1 | CPU | 3.6 | _ | _ | _ | Arch Linux x86_64 | Download |
| 1.13.1 | CPU | 3.6 | _ | _ | SSE4.X,AVX,AVX2,FMA | Ubuntu 18.04 x86_64 | Download |
| 1.13.0 | GPU | 3.7 | 10.0 | 7.4 | XLA,SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 1.13.0 | GPU | 3.7 | 10.0 | 7.4 | SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 1.13.0 | CPU | 3.7 | _ | _ | SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| 1.12.0 | GPU | 3.6 | 10.0 | 7.4 | SSE4.X,AVX,AVX2,FMA | Arch Linux x86_64 | Download |
| TF | HW | Python | CUDA | cuDNN | Support | OS | Download |
|---|---|---|---|---|---|---|---|
| 2.6.1 | CPU / skylake-512 | 3.7 | _ | _ | AVX512F | CentOS 7.x | Download |
| 2.3 | GPU CUDA Compute 5.0 | 3.8 | 11 | 8 | SSE4.X,AVX,AVX2, FMA, XLA | Arch Linux x86_64 / Ubuntu 20.04 | Download |
| 2.2 | GPU CUDA Compute 5.0 | 3.8 | 10.2 | 7.6 | SSE4.X,AVX,AVX2, FMA, XLA | Arch Linux x86_64 / Ubuntu 20.04 | Download |
| 2.1 | CPU / amdfam10 | 3.7 | _ | _ | SSE4.X,AVX,AVX2, FMA, XLA | Ubuntu 18.04 x86_64 | Download |
| 1.14.1 | CPU / amdfam10 | 3.6 | _ | _ | XLA | Ubuntu 18.04 x86_64 | Download |
| 1.13.1 | CPU / amdfam10 | 3.6 | _ | _ | _ | Ubuntu 18.04 x86_64 | Download |
| 1.13.1 | CPU / skylake-512 | 3.7 | _ | _ | SSE4.X, FMA, AVX512F | Ubuntu 18.04 x86_64 | Download |
This is a rolling distro, so you might be tempted to always upgrade everything (nvidia drivers, linux kernel, etc). One thing to take into account is that different CUDA versions have different nvidia driver requirements (and certain driver versions depend on specific kernel versions). Before upgrading the kernel or drivers, consider checking the CUDA toolkit release notes: https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
I'm using this as a repository of TF wheels I'll be using in various projects, if you want a particular build and don't have the reasources to build from sources yourself, open an issue and I'll see if I can help.
You can also drop me a line at @davidelnunes
Since I'm building on a machine with ArchLinux, which is a rolling distro. This means wheels might be compiled with e.g. glibc on a newer version than your current system. To speed-up the process, if you are requesting a custom build, providing a docker image I can use with your exact system requirements is most appreciated 👍
If you find any of this useful, consider being my sponsor. I really appreciate it, or get me some coffee, coffee is great!
