win10下如何安装GPU版本的TensorFlow

这篇文章主要为大家展示了“win10下如何安装GPU版本的TensorFlow”,内容简而易懂,条理清晰,希望能够帮助大家解决疑惑,下面让小编带领大家一起研究并学习一下“win10下如何安装GPU版本的TensorFlow”这篇文章吧。

创新互联建站致力于互联网品牌建设与网络营销,包括成都网站设计、成都做网站、SEO优化、网络推广、整站优化营销策划推广、电子商务、移动互联网营销等。创新互联建站为不同类型的客户提供良好的互联网应用定制及解决方案,创新互联建站核心团队10余年专注互联网开发,积累了丰富的网站经验,为广大企业客户提供一站式企业网站建设服务,在网站建设行业内树立了良好口碑。

一、 明确自己电脑的适配版本的cuda

在此之前你需要在电脑上安装有VS,在这里直接给出百度云连接:

链接:https://pan.baidu.com/s/1M94HKU6O47G7SGq8xYS6Pg 
提取码:9ofd 

win10下如何安装GPU版本的TensorFlow

如何确定自己的电脑型号:

右键我的电脑——>点击属性——>系统——>进入设备管理器——>找到显示适配器:

如下图方框内的就是我的nvidia型号

win10下如何安装GPU版本的TensorFlow

然后你需要找出与你的版本对应的cuda

查看一下自己电脑上有没有NVIDIA控制面板, 如果没有最好安一个(前提是你的电脑有n卡)

利用驱动精灵看一下是否有NVIDIA驱动(驱动精灵安装包链接:https://pan.baidu.com/s/12AjwmiTpT50k6wNGj4CcbQ 
提取码:wka0 )

win10下如何安装GPU版本的TensorFlow

利用驱动精灵检查一下自己的NVIDIA驱动是否为最新的,最好升级一下

是最新的就打开NVIDIA控制面板——>设置physx配置——>组件,可以看到NVIDIA.DLL, 后面的就是你应该安装的版本

win10下如何安装GPU版本的TensorFlow

win10下如何安装GPU版本的TensorFlow

二、下载对应版本的cuda, cudnn

(cuda以及cudnn版本以及对应版本补丁)

百度云:链接:https://pan.baidu.com/s/195x-Vn2-_HtI54M93cvJTQ  提取码:kqcr 

下载对应版本的cuda 

cuda下载地址:https://developer.nvidia.com/cuda-toolkit-archive

win10下如何安装GPU版本的TensorFlow

win10下如何安装GPU版本的TensorFlow

cudnn下载地址:https://developer.nvidia.com/rdp/cudnn-download(在下载时需要先注册)

win10下如何安装GPU版本的TensorFlow

三、安装cuda, cudnn

下载后安装cuda:

可以参考:https://blog.csdn.net/u013165921/article/details/77891913

安装就很简单了,同意并继续,一直下去,环境我选择的是自定义安装。

安完之后再控制面板会看到:

win10下如何安装GPU版本的TensorFlow

然后解压与cuda对应的cudnn:

win10下如何安装GPU版本的TensorFlow

将以上的文件移到 你的cuda安装目录下 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

win10下如何安装GPU版本的TensorFlow

这样就算是完成了cuda和cudnn

四、检验CUDA,CUDNN是否安装成功

win+R——>输入nvidia-smi——>nvcc -V出现以下:

win10下如何安装GPU版本的TensorFlow

(大部分情况是好了,但是我在第一次安装的时候,也是这样,但是就是报DLL错误,很烦躁了)

所以为了精确在准确的确认是否真的安装好了:

cd到根目录——>在 cd   C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\demo_suite——>分别输入deviceQuery.exe 和 bandwidthTest.exewin10下如何安装GPU版本的TensorFlow

win10下如何安装GPU版本的TensorFlow

win10下如何安装GPU版本的TensorFlow

在最好看到的都是pass那么就说明CUDA和CUDNN安装好了

五、安装TensorFlow-GPU版本

好了,安装完cuda之后就该安装TensorFlow-GPU版本的了

我的是在anconda3 + python3.6.2 + tensorflow-gpu 1.13.1版本  

# 创建一个名为tensorflow-py36的环境,指定Python版本是3.6)conda create --name tensorflow-py36 python=3.6# 安装好后,使用activate激活某个环境activate tensorflow-py36  # 即系统已经切换到了3.4的环境deactivate tensorflow-py36conda remove --name tensorflow-py36 --all

如下,不报错应该就好了,但是!!!我有一次安装的时候,这样导入是好的,但是在运行代码的时候出现了错误!!!,找不到TensorFlow中的方法。。。。。。。。。。。。

此时可能的错误就是你的项目名称和TensorFlow相同了~~~~(此处是有可能!!!)

win10下如何安装GPU版本的TensorFlow

所以为了真正的确定,tensorflow真的 是安装好了,就拿一个代码测试一下:

测试tensorflow代码:

# -*- coding: utf-8 -*-# !/usr/bin/env python# @Time    : 2019/5/7 17:19# @Author  : xhh# @Desc    :  # @File    : tensor_tf.py# @Software: PyCharmimport numpy as npimport tensorflow as tf
matrix1 =  tf.constant([[3., 3.]])matrix2 =  tf.constant([[2.],[ 2.]])
product = tf.matmul(matrix1, matrix2)print(product)sess =  tf.Session()

 代码结构没有报错~~~~

最后见证结果的时候到了,有点小激动啦~~

win10下如何安装GPU版本的TensorFlow

以上是“win10下如何安装GPU版本的TensorFlow”这篇文章的所有内容,感谢各位的阅读!相信大家都有了一定的了解,希望分享的内容对大家有所帮助,如果还想学习更多知识,欢迎关注创新互联行业资讯频道!


当前标题:win10下如何安装GPU版本的TensorFlow
转载来于:http://cdiso.cn/article/jejgih.html

其他资讯