TensorFlow有GPU版和CPU版。
GPU版需要CUDA和cuDNN支持,到链接:https://developer.nvidia.com/cuda-gpus 确认自己的显卡是否支持CUDA。本人采用pip方式安装GPU版本的TensorFlow。
安装Python:
官网下载Python:https://www.python.org/downloads/windows/,默认设置安装。
安装TensorFlow:
#GPU版本:
pip3 install --upgrade tensorflow-gpu
#CPU版本:
pip3 install --upgrade tensorflow
安装过程由于网络原因会出现错误,一般为红色。多试几次就可以安装成功。
安装CUDA:
根据官方安装:https://www.tensorflow.org/install/install_windows提示,CUDA版本为8.0,cuDNN版本为v6或者v6.1,CUDA的下载链接:https://developer.nvidia.com/cuda-toolkit-archive,选择相应的版本。
安装cuDNN:
官网链接:https://developer.nvidia.com/rdp/cudnn-download下载对应CUDA版本的cuDNN,解压添加路径(例如:D:\STORE\cuda\bin)到path变量
若官网维护可到百度云下载,链接:http://pan.baidu.com/s/1o7NM3Z8 密码:40qx
测试结果:
。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。
GPU版本的是安装成功了,仔细看看终端里面的文字: Ignoring visible gpu device (device: 0, name: GeForce 610M, pci bus id: 0000:01:00.0) with Cuda compute capability 2.1. The minimum required Cuda capability is 3.0.大致意思说,忽略了cuda的计算能力是2.1版本的GPU,需要的cuda的计算能力的最低版本为3.0.所以我的显卡太烂了 ,用不了。随后附上显卡支持的CUDA版本链接: