Windows10下Tensorflow启用GPU加速，显卡GTX1060，踩坑记录

因为需要用到tensorflow学习深度学习，所以有N卡就想开启GPU加速，结果各种坑

 

1.安装VS和Python环境  （不用VS的可以不安装，使用其他工具也是一样的）

这里使用VS2019作为开发工具！！！

记住，Python环境版本不要超过3.6，不要超过3.6，不要超过3.6

 

tensorflow 版本都有对应其他版本的，这里Python版本使用3.6.x（这个很重要，我用3.7结果下载的tensorflow版本不一样）

 

2.安装cuda9.0

 

CUDA下载地址

 

测试是否安装成功，打开cmd,输入nvcc -V 

 

 

3.安装cudnn

根据cuda9.0的版本，我们选择cudnn7.1,到cudnn官网下载对应版本for windows10,是一个压缩包

解压出来，把解压的文件拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0文件目录下，覆盖

下载地址：https://developer.nvidia.com/rdp/cudnn-download

解压后覆盖到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0目录即可。

 

4.安装pip

如果你的电脑没装pip,还要安装pip，因为我们使用pip安装tensorflow-gpu

 请自行百度

 

这是tensorflow的版本，其他版本可能会出现问题

 

pip3 install tensorflow-gpu==1.12.0
pip3 install tensorflow=1.2

 

如果出现很多警告，将numpy版本降低到1.16.0

 

5.测试GPU是否启用

 

import tensorflow as tf
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' # 只显示 Error
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

 

输出结果：

 

 

其他显卡找一找匹配的就好了

 

推荐一下我的Wy博客