人如果没有梦想那和咸鱼有什么区别,即使是做咸鱼也要做最咸的那一条

—— Wy

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

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))


输出结果:






评论

用户默认头像
发表我的评论
取消评论
看不清,点击刷新验证码
  1. 用户默认头像
    文章还不错支持一下
    热搜 2020年03月24日 22:35:24 回复
  2. 用户默认头像
    文章不错非常喜欢,支持
    今日新鲜事 2020年03月07日 00:33:07 回复
  3. 用户默认头像
    文章不错非常喜欢
    热搜榜 2020年01月28日 22:06:46 回复