Location via proxy:   [ UP ]  
[Report a bug]   [Manage cookies]                
Skip to content

davidenunes/tensorflow-wheels

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

81 Commits
 
 
 
 

Repository files navigation

Find any of this useful?

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

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

Other builds

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

Using Arch Linux?

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

Need a different build?

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

Note

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 👍

Buy me a coffee

If you find any of this useful, consider being my sponsor. I really appreciate it, or get me some coffee, coffee is great!

Buy Me a Coffee at ko-fi.com