Releases: davidenunes/tensorflow-wheels
Releases · davidenunes/tensorflow-wheels
Tensorflow 2.6.1 CPU - Python 3.7 - CentOS 7 skylake-512
TF | Python | Support | OS |
---|---|---|---|
2.6.1 | 3.7 | AVX512F | CentOS 7 |
Tensorflow 2.5.0 GPU - Python 3.9 - SSE / AVX / FMA - CUDA 11.3 - cuDNN 8
TF | Python | CUDA | cuDNN | NCCL | Support | OS | glibc |
---|---|---|---|---|---|---|---|
2.5.0 | 3.9 | 11.3 | 8 | 2.7 | SSE, AVX, FMA | Arch Linux x86_64 | 2.33 |
Tensorflow 2.5.0rc3 GPU - Python 3.9 - SSE / AVX / FMA - CUDA 11 - cuDNN 8
TF | Python | CUDA | cuDNN | NCCL | Support | OS | glibc |
---|---|---|---|---|---|---|---|
2.5.0rc3 | 3.9 | 11 | 8 | 2.7 | SSE, AVX, FMA | Arch Linux x86_64 | 2.33 |
Tensorflow 2.4.1 CPU - Python 3.9 - SSE / AVX / FMA
TF | Python | CUDA | cuDNN | NCCL | Support | OS | glibc |
---|---|---|---|---|---|---|---|
2.4.1 | 3.8 | NA | NA | NA | SSE, AVX, FMA | Arch Linux x86_64 | 2.32 |
Note: the required h5py version cannot be installed in python 3.9, I recommend installing
tensorflow with the --no-deps
flag
pip install ~/tensorflow-2.4.1-cp39-cp39-linux_x86_64.whl --no-deps
and then install the following dependencies separately:
pip install astunparse~=1.6.3 google-pasta~=0.2 gast==0.3.3 flatbuffers~=1.12.0
Tensorflow 2.4.1 GPU - Python 3.8 - SSE / AVX / FMA - CUDA 11 - cuDNN 8
TF | Python | CUDA | cuDNN | NCCL | Support | OS | glibc |
---|---|---|---|---|---|---|---|
2.4.1 | 3.8 | 11.1 | 8 | 2.7 | SSE, AVX, FMA | Arch Linux x86_64 | 2.32 |
Tensorflow 2.4 GPU - Python 3.9 - SSE / AVX / FMA - CUDA 11 - cuDNN 8
TF | Python | CUDA | cuDNN | NCCL | Support | OS | glibc |
---|---|---|---|---|---|---|---|
2.4 | 3.9 | 11.1 | 8 | 2.7 | SSE, AVX, FMA | Arch Linux x86_64 | 2.32 |
Tensorflow 2.4 GPU - Python 3.8 - SSE / AVX / FMA - CUDA 11 - cuDNN 8
TF | Python | CUDA | cuDNN | NCCL | Support | OS | glibc |
---|---|---|---|---|---|---|---|
2.4 | 3.8 | 11.1 | 8 | 2.7 | SSE, AVX, FMA | Arch Linux x86_64 | 2.32 |
Tensorflow 2.4rc4 GPU - Python 3.8
TF | Python | CUDA | cuDNN | NCCL | Support | OS |
---|---|---|---|---|---|---|
2.4rc4 | 3.8 | 11.1 | 8 | 2.7 | SSE, AVX, FMA, TensorRT | Arch Linux x86_64 |
Tensorflow 2.3 GPU - Python 3.8 - CUDA 11 - cuDNN 8 (Compute Capability 5)
Wheel for compute capability 5.0
TF | Python | CUDA | cuDNN | NCCL | Support | OS |
---|---|---|---|---|---|---|
2.3 | 3.8 | 11 | 8 | 2.7 | SSE, AVX, FMA | Arch Linux x86_64 |
Tensorflow 2.3 GPU - Python 3.8 - CUDA 11 - cuDNN 8
TF | Python | CUDA | cuDNN | NCCL | Support | OS |
---|---|---|---|---|---|---|
2.3 | 3.8 | 11 | 8 | 2.7 | SSE, AVX, FMA | Arch Linux x86_64 |