概述
前些日子在 win10 下安装TensorFlow 1.2 GPU版本,但是目前在跑一些官方demo的时候发现1.2版本的不支持了。由于电脑自带GTX1051 Ti 想充分利用资源就安装了一个前几天最新发布的1.6 GPU版本TensorFlow。
首先,你得可以翻墙,不然后面安装下载要用的东西会很慢。。。嗯,翻墙我就不多说了吧。
既然,能翻墙,那么我们就开始吧。
1,下载NVIDIA CUDA Toolkit:
连接传送门:https://developer.nvidia.com/cuda-90-download-archive
选择安装版本:嗯,看下图。。。。。我选的是深绿色的选项哦~~~~~~
安装类型选择local 下载到本地
然后弹出下载的东东,见图
刚好是第一个,看到了吧?1.4G 挺大的,网速不给力的就等等吧,然后不能翻墙,基本就卡在这里了。。
(如果是其他系统,自己选择吧)
下载完NVIDIA CUDA Toolkit就安装它,(最好路径啥的都用英文,这是个好习惯,能省很多bug)
安装过程稍微有点久。。。。。。。估计十几分钟?????
如果以前下载并安装过较早版本的 NVIDIA CUDA Toolkit 不用管直接安装,安装过程中会自动卸载忽略旧版本。
安装完过后可能会叫你重启电脑,你就重启吧。。。。。。。嗯~~~~~~
2,安装cuDNN 7.x
连接传送门:https://developer.nvidia.com/cudnn
往下 滑 有个绿色的大 Download 按钮点进去。(这需要成员才能登录,注册个账号就行了。。)
这里有一堆版本供你选择 ,1.6GPU 版本的TensorFlow 就选这个了 (见下图):
然后你在选择你系统对应的版本即可(见下图)
版本号格式:cudnn-9.0-windows[your version]-x64-v7.zip
下载完了继续:
解压你下载好的文件,等到一个cuda文件夹,进入这个cuda文件夹把这个文件夹里的东西都复制,然后拷贝到C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0文件夹下面。
看看cuda文件夹里面的东西,见下图:(所有拷贝额~~~)
3,下载whl文件
连接传送门:https://pypi.python.org/pypi/tensorflow-gpu
找到对应的版本,我python版本是3.6的,所以我选择了:
tensorflow_gpu-1.6.0-cp36-cp36m-win_amd64.whl (md5) 下图最后一个
嗯,到这里就快完成了。
运行命令:pip install [whl file path] 路径我就不多说了吧?
===================分割线 ====================
到这里,就恭喜你完成了安装了。。检查安装是够成功。。
运行一下命令:
import TensorFlow as tf
tf.__vresion__ 就可以看到你当前的版本了
最后
以上就是活泼导师为你收集整理的window 10 下tensorflow 1.6 GPU版本的安装方法的全部内容,希望文章能够帮你解决window 10 下tensorflow 1.6 GPU版本的安装方法所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
发表评论 取消回复