1. 安装cuda9.0时,如果安装失败,请下载 DDUddu下载,解压运行卸载之前的显卡驱动,自动重启后安装CUDA即可。
从官网下载cuda9.0以及相应的两个patch包,双击进行安装。(安装过程中请关闭360等杀毒软件等)
添加环境变量:
2. cudnn:
下载cudnn的相应版本,解压,将bin路径添加到环境变量
3. pip install tensorflow-gpu
4. pip list检查是否安装成功!!!
5. 用代码检查是在用CPU运行还是GPU
https://learningtensorflow.com/lesson10/