admin 管理员组

文章数量: 1086866

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文件目录下，覆盖

下载地址：

解压后覆盖到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博客

 

本文标签： Windows10下Tensorflow启用GPU加速，显卡GTX1060，踩坑记录

版权声明：本文标题：Windows10下Tensorflow启用GPU加速，显卡GTX1060，踩坑记录 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.roclinux.cn/b/1687942647a160270.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。