我是win7,显卡GTX1050Ti,选的cuda10.0,cuDNN7.4
tensorflow分为CPU版本和GPU版本,CPU版本运行较慢。
1 安装anaconda3(python3.6)
2 安装tensorflow
2.1 CPU版本安装
不依赖于其他,直接在cmd中 conda install tensflow
2.2 GPU版本安装
1)是否支持GPU版本
在控制面板-设备管理-显示配置器中查看是否有支持GPU的显卡
2)安装cuda
cuda最新(10.0)版本:https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=7&target_type=exelocal
选择适合电脑系统win7的cuda安装包:
如果需要其他版本的cuda:https://developer.nvidia.com/cuda-toolkit-archive
安装步骤一步一步来就行。
3)安装cuDNN
cuDNN版本需要与cuda匹配
https://developer.nvidia.com/rdp/cudnn-download
如果需要其他版本的cuDNN:https://developer.nvidia.com/rdp/cudnn-archive?tdsourcetag=s_pctim_aiomsg
解压后,是这些:
将这三个复制至cuda的安装目录下就可以了。
4)安装tensorflow
conda install tensflow-gpu