您当前的位置:首页 > 计算机 > 软件应用 > 开发工具(IDE)

win10安装conda+tensorflow+cpu and gpu基本流程

时间:03-12来源:作者:点击数:

1、安装anaconda,很简单直接官网下载即可,安装完成,配置环境变量,系统变量path中添加。

2、启动anaconda prompt,创建两个python环境,一个为tensorflow(安装cpu),另一个tensorflow-gpu,激活环境,在各自环境中安装python,版本为3.6较好。

3、cpu版本,直接在tensorflow环境中安装想要的版本,记得使用镜像源,速度更快,后续可继续安装keras等。

4、gpu版本,首先做准备工作,查看自己显卡的型号,可支持的cuda版本,以及cuda所对应的cudnn,算力需要>3.0,比如去英伟达驱动官网查看,确定所要安装的tensorflow-gpu、cuda、cudnn版本,一定要匹配,下载cuda,安装时自定义安装,去掉不要的选项,记住安装目录,完成安装。下载cudnn,解压到cuda文件中去。之后同样配置系统变量,配置完成,测试cuda是否成功安装。最后在tensorflow-gpu环境中安装tensorflow-gpu,完成安装,测试是否安装成功。

tensorflow版本与keras版本不匹配问题:

更新tensorflow:更新到1.6.0版本,用豆瓣镜像,加–user由于权限问题

pip install --user --upgrade --ignore-installed tensorflow==1.6.0 -i https://pypi.douban.com/simple
方便获取更多学习、工作、生活信息请关注本站微信公众号城东书院 微信服务号城东书院 微信订阅号
推荐内容
相关内容
栏目更新
栏目热门