由于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
安装完成后在命令提示符中输入Python
如果看到相应的版本,说明Anaconda已安装好
二、下载安装CUDA和cuDNN
CUDA下载地址: https://developer.nvidia.com/cuda-downloads
cuDNN下载地址:https://developer.nvidia.com/cudnn
选择CUDA版本,注意官网提供的CUDA都是9.1版本的,应当下载CUDA8.0和相应的cuDNN
cuDNN下载之前需要注册一下,不过很简单。
cuDNN安装只需要把它解压然后把解压后的文件全部复制粘贴到CUDA下即可
在命令提示符下输入nvcc –V可查看CUDA版本
三、安装Tensorflow-gpu
在命令提示符下输入pip insall 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:“
删除setuptools-27.2.0-py3.5.egg
2.打开AnacondaPrompt,输入conda
3. 联网情况下输入conda update conda pip,更新pip
在解决了这个问题后,再次输入pip insall tensorflow-gpu,
Tensorflow-gpu已经可以安装了
四、安装完成后进行测试
输入代码:
import tensorflow astf
Hello =tf.constant('I love YX')
piece = tf.Session()
print =(piece.run(Hello))
可显示如下结果则证明环境已经搭建好了
版权声明:本文为博主原创文章,未经博主允许不得转载