一、首先要版本匹配,
我的电脑适合安装的GPU是cuda9.2,然后由于tensorflow与cuda,cudnn的匹配关系,具体关系,见我上篇文章。
所以我的安装平台是
cuda9.2
cudnn7.1
Keras2.1.6
tensorflow1.8.0,
但是目前tensorflow官方版本还不支持cuda9.2,所以从网址https://github.com/fo40225/tensorflow-windows-wheel/tree/master/1.8.0。下载tensorflow1.8.0,
下载tensorflow1.8.0花费了很多时间,后来从github-dev中右键快速下载成功。
二、安装
打开Anaconda Prompt,进入Anaconda命令行管理界面。
创建运行环境,输入指令:
conda create -n tensorflow-gpu python=3.6
新建一个名字叫“tensorflow-gpu”,python版本为3.6的运行环境,此环境与Anaconda中其它环境隔离。
创建完成后,输入指令:
conda activate tensorflow-gpu
激活这个环境。
输入指令 pip install 你存放下载好的tensorflow1.8.0的路径。
pip install tensorflow_gpu-1.8.0-cp36-cp36m-win_amd64.whl
三、遇到问题
报错的大概意思就是:
protobuf需要Python ‘>=3.7’,但运行的Python是3.5.5
您使用的是pip版本9.0.1,但是可以使用23.3.2版本。
你应该考虑通过’python -m pip install --upgrade pip’命令进行升级。
升级后安装成功
四,继续安装Keras
在conda activate tensorflow-gpu的环境下继续安装
pip install keras
=
=
2.1.6
验证
输入python,
然后import Keras
安装成功。