Skip to content

TensorFlow 1.4 (GPU, CUDA 9, Compute 3.7/6.0/7.0)
Compare
Choose a tag to compare
@amberfly amberfly released this 08 Nov 01:37
· 32 commits to master since this release
tf1.4-gpu-cuda9
8c300a2
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

GPU wheels for TensorFlow 1.4, 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). If you are on K80, these wheels are built only with 3.7 and are smaller.

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

Assets 5