Windows 10安装TensorFlow-gpu1.4 及CUDA8.0,cuDNN6.0,搞定了
先说下我的最终版本:
Python 3.5.3;
tensorflow-gpu==1.4.0 ;
CUDA Toolkit 8.0 GA1 (Sept 2016), Online Documentation;
cuDNN v6.0(Download cuDNN v6.0 (April 27, 2017), for CUDA 8.0)
注意CUDA的版本是由TensorFlow决定的。具体如何对应,需要去TensorFlow官网看:https://www.tensorflow.org/install/install_windows
https://www.tensorflow.org/install/install_windows#requirements_to_run_tensorflow_with_gpu_support(很抱歉,大陆打不开2018.9.10),还有个方法是,下载喜欢的TF版本,让系统报错提示你安装合适的CUDA版本
安装TensorFlow:pip install tensorflow-