由于Tesnsorflow-gpu目前在win10的环境下对其相应配套软件的版本要求较为苛刻,小编踩过很多坑,因此在此写下这份教程,希望能帮助到大家。

一、下载安装Anaconda

Anaconda官网上提供的Anaconda为3-5.0.1版本,由于该版本对应的Python为3.6,无法支持Tensorflow-gpu模块,因此应下载Anaconda3-4.2.0-Windows-x86_64.HYPERLINK"https://repo.continuum.io/archive/Anaconda3-4.2.0-Windows-x86_64.exe"exe

下载网址:https://repo.continuum.io/archive/index.html

Win10下搭建Tensorflow-gpu环境

安装完成后在命令提示符中输入Python

Win10下搭建Tensorflow-gpu环境

如果看到相应的版本,说明Anaconda已安装好

二、下载安装CUDAcuDNN

 CUDA下载地址: https://developer.nvidia.com/cuda-downloads

 cuDNN下载地址:https://developer.nvidia.com/cudnn

Win10下搭建Tensorflow-gpu环境

选择CUDA版本,注意官网提供的CUDA都是9.1版本的,应当下载CUDA8.0和相应的cuDNN

Win10下搭建Tensorflow-gpu环境

cuDNN下载之前需要注册一下,不过很简单。

Win10下搭建Tensorflow-gpu环境

cuDNN安装只需要把它解压然后把解压后的文件全部复制粘贴到CUDA下即可

在命令提示符下输入nvcc –V可查看CUDA版本

Win10下搭建Tensorflow-gpu环境

三、安装Tensorflow-gpu

在命令提示符下输入pip insall tensorflow-gpu

Win10下搭建Tensorflow-gpu环境

报错了!这是什么原因,不是本身命令的问题,而是Anaconda自带的问题,这是搭建Tensorflow环境中很多人都踩过的坑,那么如何解决呢?

分为三步:

1.找到Anaconda安装目录中的path.py,详细路径为:

C:\Users\hacker\Anaconda3\Lib\site-packages\conda\_vendor\auxlib\path.py

把文件path.py中,把“except ImportError:”修改为“except Exception:

Win10下搭建Tensorflow-gpu环境

删除setuptools-27.2.0-py3.5.egg

Win10下搭建Tensorflow-gpu环境

2.打开AnacondaPrompt,输入conda

Win10下搭建Tensorflow-gpu环境

3. 联网情况下输入conda update conda pip,更新pip

Win10下搭建Tensorflow-gpu环境

 

在解决了这个问题后,再次输入pip insall tensorflow-gpu

Win10下搭建Tensorflow-gpu环境

Tensorflow-gpu已经可以安装了

四、安装完成后进行测试

Win10下搭建Tensorflow-gpu环境

输入代码:

import tensorflow astf

Hello =tf.constant('I love YX')

piece = tf.Session()

print =(piece.run(Hello))

可显示如下结果则证明环境已经搭建好了

Win10下搭建Tensorflow-gpu环境

 

 

 

 

 

 

 

 

 

版权声明:本文为博主原创文章,未经博主允许不得转载

相关文章: