Window10安装TensorFlow(GPU)与可行性测试

来源:互联网 发布:cocos2dx mac环境搭建 编辑:程序博客网 时间:2024/05/29 17:54

2017*11*9遇到坑了,安装成功但是import tensorflow出错正在排查原因。


因为在VM的Ubuntu上貌似对GPU支持不怎么好,使用体验不佳,现在尝试直接在Windows10上使用anaconda和pip安装tensorflow。

2017年11月9日修订:TensorFlow已经放出1.2 for windows 版本。支持python3.6


先给出参考文章:

http://blog.sina.com.cn/s/blog_14935c5880102wu86.html

http://blog.csdn.net/cs_leebo/article/details/64444243

http://jingyan.baidu.com/article/c74d6000a368090f6a595d95.html

http://blog.csdn.net/u010858605/article/details/64128466


1.检查GPU是否支持CUDN​:其实这一步是可以省略的,因为CUDN支持的显卡范围特别广,你基本只要确认你的GPU是不是NVIDIA的就可以了,当然如果你还是不确定的话,也可以去这个网站(https://developer.nvidia.com/cuda-gpus)中的“CUDA-Enabled GeForce Products”查一下,只要你的显卡型号在里面就可以了。


2.下载并按照CUDA:进入此网站,(https://developer.nvidia.com/cuda-downloads),注册登录。

按照下图选择即可:

 

下载完之后打开安装,一路确认到底

​安装完后即可。

3.下载cuDNN:​其实,到了这一步比较闹心,因为当你进了下载网站(https://developer.nvidia.com/rdp/form/cudnn-download-survey)。


下载后将其解压

​在C盘根目录下新建个tools文件夹,将解压后的“cuda”文件夹放入其中


​4.设置环境变量:右键“我的电脑”,选择“高级系统设置”​进入“环境变量”进入“Path”,

(实际上我添加了这些)

C:\tools\cuda\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin\lib\x64

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0

后点击确认。

在http://blog.sina.com.cn/s/blog_14935c5880102wu86.html这篇文章的安装过程中出现了问题。后再在这个基础上参考了http://blog.csdn.net/u010858605/article/details/64128466

//////////////////////////////////////////////////////////////////////////////////////////


1.安装Anaconda


选择相应的Anaconda进行安装,

Anaconda3-5.0.1-Windows-x86_64 .exe这是我安装的版本

就和安装普通的软件一样,全部选择默认即可,注意勾选将python3.6添加进环境变量。 
这里写图片描述

这样Anaconda就安装好了,我们可以通过下面的命令来查看Anaconda已经安装了哪些包。 

在开始菜单(只要你能找到就好)找到Anaconda Prompt(强调一下,一定要是管理员打开!!!

conda list

可以看到已经安装了numpy、sympy等常用的包。


2.安装Tensorflow

TensorFlow目前在Windows下只支持Python 3.5版本。

(1)打开Anaconda Prompt(强调一下,一定要是管理员打开!!!),输入清华仓库镜像,这样更新会快一些: 
这里写图片描述

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --set show_channel_urls yes
  • 1
  • 2

(2)同样在Anaconda Prompt中利用Anaconda创建一个python3.5的环境,环境名称为tensorflow ,输入下面命令:

conda create -n tensorflow python=3.5
  • 1

运行 开始菜单->Anaconda3—>Anaconda Navigator,点击左侧的Environments,可以看到tensorflow的环境已经创建好了。

这里写图片描述

这里写图片描述

(3)在Anaconda Prompt中启动tensorflow环境:

activate tensorflow
注:当不使用tensorflow时,关闭tensorflow环境,命令为:deactivate

(4)安装TensorFlow

cpu版本的

pip install --upgrade --ignore-installed tensorflow

gpu版本的

pip install --upgrade --ignore-installed tensorflow-gpu

等待安装tensorflow就好了。(记住前面提到的用管理员,这是一个坑,要不然安装过程容易出错)








原创粉丝点击