Tensorflow 1.4 currently works with CUDA 8.0, which is poorly documented. If you don't want to worry about installing CUDA 8.0 (after upgrading Tensorflow to 9.0), you can also install tf-nightly, which works with CUDA 9.0. It may be a less stable release, but it is usually as good (if not better) than major releases.
source share