1、安装anaconda
2、新建TensorFlow-gpu环境
3、安装cuda
(1)查看gpu对应cuda版本:桌面右键->NVIDIA控制面板->帮助->系统信息->组件->查看cuda信息(图1);
(2)下载cuda并安装:cuda下载地址;(目录:C:\Program Files\NVIDIA GPU Computing Toolkit)
(3)下载cudnn:cudnn下载地址(需要账号密码登录,qq邮箱、账号annatu1206、密码含字母数字特殊字符);
下载完成后不需要安装,解压后替换掉cuda9.0根目录下的文件就行
4、进入TensorFlow-gpu环境,安装tensorflow-gpu
(1)升级pip工具防止安装到一半报错:python -m pip install --upgrade pip
(2)安装tensorflow-gpu:pip install --ignore-installed --upgrade tensorflow-gpu
遇到的问题:
1、安装tensorflow-gpu有点问题,import时报错
原因:安装的cuda10.1
解决方案:卸载cuda10.1,需保留驱动、系统软件、vs相关软件;然后重新安装cuda10.0
图1: