1. 安装cuda9.0时,如果安装失败,请下载 DDUddu下载,解压运行卸载之前的显卡驱动,自动重启后安装CUDA即可。

从官网下载cuda9.0以及相应的两个patch包,双击进行安装。(安装过程中请关闭360等杀毒软件等)

添加环境变量:

安装tensorflow-gpu+cuda9.0

2. cudnn:

下载cudnn的相应版本,解压,将bin路径添加到环境变量

3. pip install tensorflow-gpu

4. pip list检查是否安装成功!!!

5. 用代码检查是在用CPU运行还是GPU

https://learningtensorflow.com/lesson10/


相关文章: