Win10安装TensorFlow1.9-GPU版本

前言

前段时间在电脑上更新了tf1.4到1.9,没想到踩了这么多坑。 . 特地记录下来,希望对大家有帮助。

删除旧版本

如果你的电脑上没有安装旧版本的tf,可以忽略这一步。我需要先卸载旧版本,因为我想升级到最新版本。老版本使用安装,卸载很简单,先进入安装tf的环境,我的环境是””:

activate tensorflow  

然后卸载它,或者直接删除这个环境,卸载就可以使用了:

(tensorflow)pip uninstall tensorflow-gpu

图片[1]-Win10安装TensorFlow1.9-GPU版本-唐朝资源网

前提是你安装的时候使用pip命令。要直接删除环境,可以使用:

conda env remove -n tensorflow  

安装 CUDA

安装CUDA需要满足以下四个条件:

A CUDA- GPU(可以运行 CUDA 的显卡) A 的(启用 CUDA 的系统) A 的(启用 CUDA 的)CUDA (at)(CUDA 安装文件)

这里需要注意的是-gpu1.9的安装必须严格安装CUDA9.0,任何9.X都是不允许的,比如CUDA9. 2或任何东西,因为它会寻找.dll。如果是CUDA9.2之类的,这些dll的名字都是.dll,会报找不到的错误。

还有一点需要注意,如果你的电脑上没有VS,那么需要先安装,请自行下载安装,或者从我给的链接下载:

图片[2]-Win10安装TensorFlow1.9-GPU版本-唐朝资源网

2015

VS的安装需要一些时间,请耐心等待。

如果你认为满足前三个条件,那么就可以下载CUDA9.0,下载地址为CUDA9.0,如下图选择,然后下载:

下载完成后,双击安装。这里还有一个坑。我的显卡是 GTX 750M。如果我直接选择“lite”模式安装,会报错,安装失败。错误原因是安装不正确导致的失败。

你可以选择简单安装试试运气,也可以按照我下面的教程选择“自定义”安装。以下是自定义安装的方法:

图片[3]-Win10安装TensorFlow1.9-GPU版本-唐朝资源网

安装 cuDnn

也请严格按照tf官网要求下载.0版本,请勿下载7.X,下载地址cuDnn

然后解压得到3个文件夹,将这3个文件夹中的文件复制到CUDA安装路径下的对应文件夹中。 CUDA默认安装路径为:C:FilesGPUCUDAv9.0,具体为:

安装显卡

相比之前繁琐的步骤,这里的步骤是最简单的。请安装,本质上是一个包和版本管理工具。先去官网下载,然后安装,注意勾选其中一个选项添加到PATH中(虽然不推荐,但是很方便!),其他选项都可以选择默认选项,最多更改安装路径,其他默认安装。

1、用

创建一个虚拟环境

C:> conda create -n tensorflow python=3.6  

2、

中,你可以使用 conda 创建一个虚拟环境。但是,在内部,我们建议使用 pip 命令而不是 conda 命令来安装。

根据tf官方文档的建议,我们使用pip命令安装,但是同时我们没有执行tf官方文档中的安装命令,因为安装太慢了,我们使用豆瓣镜像安装最新版本 可以,瞬间安装

()C:> pip –index–gpu

3、测试

这里使用这个 IDE 进行测试。首先创建一个空项目,然后将解释器设置为.5,然后使用以下代码创建一个空文件:

作为 tf

图片[4]-Win10安装TensorFlow1.9-GPU版本-唐朝资源网

你好 = tf.(“你好!”)

sess = tf.()

打印(sess.run(hello))

如果输出如下,说明安装成功:

结论

到此为止,已经安装成功了,确实是很多坑。 . 我希望它可以帮助你。如有其他意外,请来信讨论解决办法。

原文链接:

© 版权声明
THE END
喜欢就支持一下吧
点赞7赞赏 分享
评论 抢沙发
头像
欢迎您留下宝贵的见解!
提交
头像

昵称

取消
昵称表情代码图片

    暂无评论内容